mmrotate
第一步:配置好环境
我直接在师姐配置的mmrotate环境上面跑的
第二步:
准备数据集
我的标注文件之前是xml,随后将其转换成dota格式
修改tools/data/dota/split/img_split.py里面的输入图片文件路径、标签路径和输出图片路径
运行img_split.py这个文件
第三步修改参数
今天跑的是rotated_faster_rcnn_r50_fpn_1x_dota_le90.py
所以修改./configs/rotated_faster_rcnn/rotated_faster_rcnn_r50_fpn_1x_dota_le90.py里面的num_classes为1(我只有一个vessel目标)
设置权重
后面的参数修改可以参考这位大佬,我做测试能不能跑,没有修改参数【精选】MMRotate从零开始训练自己的数据集_mmrotate训练自己数据集_江小白jlj的博客-CSDN博客
最重要的!!!!!!!debug搞了好久(这个是在mmrotate/datasets/dota.py里面哈)
这里括号里面要加个逗号才能跑通!还不理解是为啥,如果有大佬看到了方便的话还请赐教
第四步
运行python tools/train.py --config configs/rotated_faster_rcnn/rotated_faster_rcnn_r50_fpn_1x_dota_le90.py
ok结束!