mmrotate.md

mmrotate 模型支持列表

mmrotate 是一个基于 PyTorch 的旋转物体检测的开源工具箱，也是 OpenMMLab 项目的一部分。

安装 mmrotate

参照 official installation guide。

支持列表

Model	Task	ONNX Runtime	TensorRT	NCNN	PPLNN	OpenVINO	Model config
RotatedRetinaNet	RotatedDetection	Y	Y	N	N	N	config
Oriented RCNN	RotatedDetection	Y	Y	N	N	N	config
Gliding Vertex	RotatedDetection	N	Y	N	N	N	config
RoI Transformer	RotatedDetection	Y	Y	N	N	N	config

使用举例

# convert ort
python tools/deploy.py \
configs/mmrotate/rotated-detection_onnxruntime_dynamic.py \
$MMROTATE_DIR/configs/rotated_retinanet/rotated_retinanet_obb_r50_fpn_1x_dota_le135.py \
$MMROTATE_DIR/checkpoints/rotated_retinanet_obb_r50_fpn_1x_dota_le135-e4131166.pth \
$MMROTATE_DIR/demo/demo.jpg \
--work-dir work-dirs/mmrotate/rotated_retinanet/ort \
--device cpu

# compute metric
python tools/test.py \
    configs/mmrotate/rotated-detection_onnxruntime_dynamic.py \
    $MMROTATE_DIR/configs/rotated_retinanet/rotated_retinanet_obb_r50_fpn_1x_dota_le135.py \
    --model work-dirs/mmrotate/rotated_retinanet/ort/end2end.onnx \
    --metrics mAP

# generate submit file
python tools/test.py \
    configs/mmrotate/rotated-detection_onnxruntime_dynamic.py \
    $MMROTATE_DIR/configs/rotated_retinanet/rotated_retinanet_obb_r50_fpn_1x_dota_le135.py \
    --model work-dirs/mmrotate/rotated_retinanet/ort/end2end.onnx \
    --format-only \
    --metric-options submission_dir=work-dirs/mmrotate/rotated_retinanet/ort/Task1_results

注意：

• mmrotate 模型需要额外输入，但我们无法直接获取它。在导出模型时，可以使用 $MMROTATE_DIR/demo/demo.jpg 作为输入。