Releases: bubbliiiing/faster-rcnn-pytorch
Releases · bubbliiiing/faster-rcnn-pytorch
v3.1 -step、cos学习率下降、多GPU训练、种类目标数量计算等
重要更新
- 修复了类别转换的问题,修复了rois数量不对齐的问题
- 支持step、cos学习率下降法。
- 支持adam、sgd优化器选择。
- 支持不同预测模式的选择,单张图片预测、文件夹预测、视频预测、图片裁剪、各个种类目标数量计算。
- 增加了多GPU训练。
v2.0 -添加了大量的注释、可调整参数等
重要更新
- 更新train.py文件,增加了大量的注释,增加多个可调整参数。
- 更新predict.py文件,增加了大量的注释,增加fps、视频预测、批量预测等功能。
- 更新frcnn.py文件,增加了大量的注释,增加先验框选择、置信度、非极大抑制等参数。
- 合并get_dr_txt.py、get_gt_txt.py和get_map.py文件,通过一个文件来实现数据集的评估。
- 更新voc_annotation.py文件,增加多个可调整参数。
- 更新summary.py文件,用于观看网络结构。
Weights of Faster Rcnn
Weights of Faster Rcnn