MMrotate1.x 生成DOTA数据集提交文件的完整指南

概述

MMrotate1.x 是一个基于PyTorch的开源旋转目标检测框架，广泛应用于遥感图像分析领域。在使用MMrotate进行DOTA数据集测试时，生成符合官方要求的提交文件是评估模型性能的关键步骤。本文将详细介绍如何在MMrotate1.x中正确生成提交到DOTA服务器的压缩文件。

生成提交文件的核心命令

在MMrotate1.x中，生成DOTA格式提交文件的核心命令如下：

python tools/test.py \
    configs/h2rbox/h2rbox-le90_r50_fpn_adamw-1x_dota.py \
    work_dirs/h2rbox-le90_r50_fpn_adamw-1x_dota/epoch_12.pth \
    --work-dir=work_dirs/Task1_results

命令参数解析

1. 配置文件路径：configs/h2rbox/h2rbox-le90_r50_fpn_adamw-1x_dota.py

   • 指定了模型架构和训练配置

2. 模型权重文件：work_dirs/h2rbox-le90_r50_fpn_adamw-1x_dota/epoch_12.pth

   • 训练好的模型权重文件路径

3. 输出目录：--work-dir=work_dirs/Task1_results

   • 指定结果保存路径，DOTA格式的提交文件将生成在此目录下

生成文件结构

执行上述命令后，系统会在指定目录下生成符合DOTA评估服务器要求的文件结构：

work_dirs/Task1_results/
├── results.pkl
└── Task1/
    ├── P0006.txt
    ├── P0007.txt
    ├── ...
    └── Pxxxx.txt

其中，每个.txt文件对应一张测试图像的检测结果，格式符合DOTA官方要求。

常见问题解决方案

1. 版本兼容性问题：

   • 确保使用的MMrotate版本与配置文件匹配
   • 检查PyTorch和CUDA版本是否兼容

2. 文件格式错误：

   • 验证生成的.txt文件是否符合DOTA格式规范
   • 检查坐标值是否在合理范围内

3. 性能评估差异：

   • 确认测试时使用的数据增强方式与训练一致
   • 检查NMS阈值等后处理参数设置

最佳实践建议

1. 批量测试：可以编写脚本批量测试多个模型，比较不同配置的性能差异

2. 结果验证：在正式提交前，建议使用DOTA官方提供的评估工具本地验证结果文件

3. 版本控制：记录每次提交对应的模型配置和训练参数，便于结果复现和分析

通过以上步骤，用户可以顺利生成符合DOTA服务器要求的提交文件，完成模型性能的官方评估。