mmdetection-虚拟环境的搭建与代码调试

1、mmdetection---虚拟环境配置

官方要求：

初始环境：

• python==3.6.8
• CUDA==9.0
• Pytorch==1.0.0
• GCC==4.8.4
• mmcv>=0.2.6

遇到的问题：

①、Segmentation fault(core dumped)-段错误

使用‘‘gdb’’追踪错误：

错误原因：gcc和系统版本太低，需要升级系统和gcc版本;

参考网址：https://github.com/open-mmlab/mmdetection/issues/24

②、Cuda版本与RTX2080有冲突：THCudaCheck FAIL：

 

错误原因：显卡使用RTX 2080Ti的时候，CUDA就要装10以上。还没有验证......

③、AttributeError: model 'torch.nn' has no attribute 'SyncBatchNorm'；

解决方法：升级Pytorch版本至Pytorch==1.1.0。

（有关Pytorch安装较慢的问题解决，可参考：https://blog.csdn.net/songchunxiao1991/article/details/95192063）

最后，实现的实践环境：

 

• python==3.6.8
• CUDA==9.0+
• Pytorch==1.1.0（如果是Pytorch==1.0.0，会报错：SyncBatchNorm）
• GCC==5.4.0
• mmcv>=0.2.6

 

2、mmdetection---代码调试

调试的数据格式是采用VOC的格式进行的。

①、输入数据的修改：./mmdet/datasets：-voc.py（图片文件默认是*.jpg，如若不是，对应源码进行修改）

Note：CLASSES修改为：自己检测的LabelName

②、配置文件的修改：./configs/cascade_rcnn_r50_fpn_x1.py

Note：num_classes修改为目标检测LabelNum+1

Note：batch_size/epochs/model_save/shape都在config配置文件中进行修改

Note：当gpu数量为8时,lr=0.02；当gpu数量为4时,lr=0.01；只要一个gpu，所以设置lr=0.0025

③、评估阶段的修改：./mmdet/core/evaluation/class_names.py

Note：修改cov_classes为自己的检测LabelNames（与第一步操作的LabelName一致）

3、训练与验证的代码操作

①、训练代码操作：

python tools/train.py configs/cascade_rcnn_r50_fpn_1x.py

Note：前提：训练、验证、测试数据已经提前写在模型的配置文件中（见2.2的配置文件的修改环节）

 

②、测试代码操作

• 首先，VOC格式数据需要先生成对应的结果*.pkl

python tools/test.py configs/cascade_rcnn_r50_fpn_1x.py work_dirs/cascade_rcnn_r50_fpn_1x/epoch_50.pth --out Result_file/Res_test.pkl

• 然后，使用voc_eval.py评估指标

python tools/voc_eval.py Result_file/Res_test.pkl configs/TB_cascade_rcnn_r50_fpn_1x.py

结果展示：

因为项目需要关注真阳和假阳的指标，因此在源码的基础上添加了tp和fp的指标：