MMRotate：基于 PyTorch 的旋转目标检测基准工具箱
计算机视觉与模式识别
2022-07-20 v4 人工智能
摘要
我们提出了一个名为 MMRotate 的开源工具箱，其为基于深度学习的流行旋转目标检测算法提供了训练、推理与评估的一致算法框架。MMRotate 实现了 18 种最先进（SOTA）算法，并支持三种最常用的角度定义方法。为了促进旋转目标检测相关问题的未来研究与工业应用，我们还提供了大量训练好的模型与详细基准，以揭示旋转目标检测的性能表现。MMRotate 已在 https://github.com/open-mmlab/mmrotate 公开发布。
引用
@article{arxiv.2204.13317,
title = {MMRotate: A Rotated Object Detection Benchmark using PyTorch},
author = {Yue Zhou and Xue Yang and Gefan Zhang and Jiabao Wang and Yanyi Liu and Liping Hou and Xue Jiang and Xingzhao Liu and Junchi Yan and Chengqi Lyu and Wenwei Zhang and Kai Chen},
journal= {arXiv preprint arXiv:2204.13317},
year = {2022}
}
备注
5 pages, 2 tables, MMRotate is accepted by ACM MM 2022 (OS Track). Yue Zhou and Xue Yang provided equal contribution. The code is publicly released at https://github.com/open-mmlab/mmrotate