【mmrotate】旋转目标检测之训练DOTA数据集

every blog every motto: You can do more than you think.
https://blog.csdn.net/weixin_39190382?type=blog

0. 前言

mmrotate训练DOTA数据集记录

1. 正文

1.1 数据准备

数据介绍部分，参考DOTA数据介绍，官方提供了裁剪工具development kit。这里不使用官方的，使用mmrotate里面的裁剪工具。

首先将数据下载并解压放到DOTA文件夹中（在mmrotate下新建data文件夹）,文件夹形式如下：

mmrotate
├── mmrotate
├── tools
├── configs
├── data
│ ├── DOTA
│ │ ├── train
│ │ ├── val
│ │ ├── test

1.1.1 数据裁剪

裁剪成1024* 1024,重叠200像素。

python tools/data/dota/split/img_split.py --base-json \
  tools/data/dota/split/split_configs/ss_trainval.json

python tools/data/dota/split/img_split.py --base-json \
  tools/data/dota/split/split_configs/ss_test.json

裁剪以后：
在这里插入图片描述

记得修改configs/_base_/datasets/dotav1.py里面的data_root!
在这里插入图片描述

关于数据处理部分，可参考mmrotate项目里面的tools/data/dota/README.md

1.2 训练

在config中选择一个配置，如下代码所示即可

python ./tools/train.py configs/rotated_retinanet/rotated_retinanet_obb_r50_fpn_1x_dota_le90.py

参考

[1] https://blog.csdn.net/weixin_39190382/article/details/129027401?spm=1001.2014.3001.5502