记录跑代码的日常(个人跑mmrotate操作)

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结束!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值