mmrotate学习（4）:mmrotate框架训练数据集

搭建环境

制作数据集

开始训练

主要是修改参数

1.修改tools/tain.py

两个主要的参数：
1）config: 使用的模型文件；
2）work-dir：训练得到的模型及配置信息保存的路径；

2.修改选用的config模型文件的参数

两个主要的参数：
1）类别数， 我这里改成num_classes = 1
2）设置预训练权重的地址，修改为你下载的预训练权重地址。

3.同时修改mmrotate/datasets/dota.py 文件中的类别名字

一个主要的参数：
1）类别名字， 我这里改成CLASSES = (‘ship’, )

4.修改configs/base/datasets/dotav1.py

两个主要的参数：
1）数据集路径
2）batchsize

5.修改configs/base/schedules/schedule_1x.py

一个主要的参数：
1）学习率lr

6.修改configs/base/default_runtime.py

也可以不修改

7.输入指令

python train.py