├── .gitignore ├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── uav_bd.py │ │ └── uav_bd_coco_mix.py │ ├── models │ │ └── yoloRm.py │ └── schedules │ │ └── yoloR_schedule.py └── uav_bd │ ├── yoloRm_uav_bd.py │ ├── yoloRm_uav_bd_coco_mix.py │ ├── yoloRm_uav_bd_coco_mix_ddp.py │ └── yoloRm_uav_bd_ddp.py ├── mmdetrot ├── __init__.py ├── core │ ├── __init__.py │ ├── anchor │ │ ├── __init__.py │ │ └── yolo_gaussianR_anchor_generator.py │ ├── bbox │ │ ├── __init__.py │ │ ├── assigner │ │ │ └── __init__.py │ │ ├── coder │ │ │ ├── __init__.py │ │ │ └── gaussianR_coder.py │ │ ├── iou_calculators │ │ │ ├── __init__.py │ │ │ └── iou2dR_calculator.py │ │ ├── samplers │ │ │ └── __init__.py │ │ └── transforms.py │ ├── custom_hooks │ │ ├── __init__.py │ │ ├── accum_optim_hooks.py │ │ ├── ema_hooks.py │ │ └── warmup_hooks.py │ ├── evaluation │ │ ├── __init__.py │ │ └── mean_apR_flexible.py │ ├── post_processing │ │ ├── __init__.py │ │ └── bbox_nmsR.py │ └── visualization │ │ ├── __init__.py │ │ └── image.py ├── datasets │ ├── __init__.py │ ├── enable_mosaic.py │ ├── pipelines │ │ ├── __init__.py │ │ └── transforms.py │ ├── uav_bd.py │ └── uav_bd_bgmix.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ └── darknetcsp.py │ ├── dense_heads │ │ ├── __init__.py │ │ ├── yoloR_head.py │ │ └── yolocsp_head.py │ ├── detectors │ │ ├── __init__.py │ │ ├── showR_mixin.py │ │ └── single_stageR.py │ ├── losses │ │ ├── __init__.py │ │ └── gaussian_distance_loss.py │ └── necks │ │ ├── __init__.py │ │ └── yolo_neck_csp.py ├── ops │ ├── __init__.py │ ├── eval_utils │ │ ├── iou │ │ │ └── iou_coco.pyx │ │ └── match │ │ │ └── match_coco.pyx │ └── geo │ │ ├── rbox_geo.cpp │ │ └── rbox_geo_utils.hpp └── version.py ├── requirements ├── build.txt ├── docs.txt ├── mminstall.txt ├── optional.txt ├── readthedocs.txt ├── runtime.txt └── tests.txt ├── setup.py ├── tests ├── test_rbbox_iou.py ├── test_uav_bd.py └── test_yoloR_anchor_generator.py └── tools ├── data_preprocess └── uav_bd_make_ids.py ├── dist_test.sh ├── dist_train.sh ├── misc └── browse_dataset.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/_base_/datasets/uav_bd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/configs/_base_/datasets/uav_bd.py -------------------------------------------------------------------------------- /configs/_base_/datasets/uav_bd_coco_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/configs/_base_/datasets/uav_bd_coco_mix.py -------------------------------------------------------------------------------- /configs/_base_/models/yoloRm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/configs/_base_/models/yoloRm.py -------------------------------------------------------------------------------- /configs/_base_/schedules/yoloR_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/configs/_base_/schedules/yoloR_schedule.py -------------------------------------------------------------------------------- /configs/uav_bd/yoloRm_uav_bd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/configs/uav_bd/yoloRm_uav_bd.py -------------------------------------------------------------------------------- /configs/uav_bd/yoloRm_uav_bd_coco_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/configs/uav_bd/yoloRm_uav_bd_coco_mix.py -------------------------------------------------------------------------------- /configs/uav_bd/yoloRm_uav_bd_coco_mix_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/configs/uav_bd/yoloRm_uav_bd_coco_mix_ddp.py -------------------------------------------------------------------------------- /configs/uav_bd/yoloRm_uav_bd_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/configs/uav_bd/yoloRm_uav_bd_ddp.py -------------------------------------------------------------------------------- /mmdetrot/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/anchor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/anchor/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/anchor/yolo_gaussianR_anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/anchor/yolo_gaussianR_anchor_generator.py -------------------------------------------------------------------------------- /mmdetrot/core/bbox/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/bbox/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/bbox/assigner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/bbox/assigner/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/bbox/coder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/bbox/coder/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/bbox/coder/gaussianR_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/bbox/coder/gaussianR_coder.py -------------------------------------------------------------------------------- /mmdetrot/core/bbox/iou_calculators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/bbox/iou_calculators/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/bbox/iou_calculators/iou2dR_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/bbox/iou_calculators/iou2dR_calculator.py -------------------------------------------------------------------------------- /mmdetrot/core/bbox/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/bbox/samplers/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/bbox/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/bbox/transforms.py -------------------------------------------------------------------------------- /mmdetrot/core/custom_hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/custom_hooks/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/custom_hooks/accum_optim_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/custom_hooks/accum_optim_hooks.py -------------------------------------------------------------------------------- /mmdetrot/core/custom_hooks/ema_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/custom_hooks/ema_hooks.py -------------------------------------------------------------------------------- /mmdetrot/core/custom_hooks/warmup_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/custom_hooks/warmup_hooks.py -------------------------------------------------------------------------------- /mmdetrot/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/evaluation/mean_apR_flexible.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/evaluation/mean_apR_flexible.py -------------------------------------------------------------------------------- /mmdetrot/core/post_processing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/post_processing/__init__.py -------------------------------------------------------------------------------- /mmdetrot/core/post_processing/bbox_nmsR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/post_processing/bbox_nmsR.py -------------------------------------------------------------------------------- /mmdetrot/core/visualization/__init__.py: -------------------------------------------------------------------------------- 1 | from .image import imshowR_det_bboxes 2 | -------------------------------------------------------------------------------- /mmdetrot/core/visualization/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/core/visualization/image.py -------------------------------------------------------------------------------- /mmdetrot/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/datasets/__init__.py -------------------------------------------------------------------------------- /mmdetrot/datasets/enable_mosaic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/datasets/enable_mosaic.py -------------------------------------------------------------------------------- /mmdetrot/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /mmdetrot/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /mmdetrot/datasets/uav_bd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/datasets/uav_bd.py -------------------------------------------------------------------------------- /mmdetrot/datasets/uav_bd_bgmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/datasets/uav_bd_bgmix.py -------------------------------------------------------------------------------- /mmdetrot/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/__init__.py -------------------------------------------------------------------------------- /mmdetrot/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/backbones/__init__.py -------------------------------------------------------------------------------- /mmdetrot/models/backbones/darknetcsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/backbones/darknetcsp.py -------------------------------------------------------------------------------- /mmdetrot/models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /mmdetrot/models/dense_heads/yoloR_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/dense_heads/yoloR_head.py -------------------------------------------------------------------------------- /mmdetrot/models/dense_heads/yolocsp_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/dense_heads/yolocsp_head.py -------------------------------------------------------------------------------- /mmdetrot/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/detectors/__init__.py -------------------------------------------------------------------------------- /mmdetrot/models/detectors/showR_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/detectors/showR_mixin.py -------------------------------------------------------------------------------- /mmdetrot/models/detectors/single_stageR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/detectors/single_stageR.py -------------------------------------------------------------------------------- /mmdetrot/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/losses/__init__.py -------------------------------------------------------------------------------- /mmdetrot/models/losses/gaussian_distance_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/losses/gaussian_distance_loss.py -------------------------------------------------------------------------------- /mmdetrot/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/necks/__init__.py -------------------------------------------------------------------------------- /mmdetrot/models/necks/yolo_neck_csp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/models/necks/yolo_neck_csp.py -------------------------------------------------------------------------------- /mmdetrot/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mmdetrot/ops/eval_utils/iou/iou_coco.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/ops/eval_utils/iou/iou_coco.pyx -------------------------------------------------------------------------------- /mmdetrot/ops/eval_utils/match/match_coco.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/ops/eval_utils/match/match_coco.pyx -------------------------------------------------------------------------------- /mmdetrot/ops/geo/rbox_geo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/ops/geo/rbox_geo.cpp -------------------------------------------------------------------------------- /mmdetrot/ops/geo/rbox_geo_utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/ops/geo/rbox_geo_utils.hpp -------------------------------------------------------------------------------- /mmdetrot/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/mmdetrot/version.py -------------------------------------------------------------------------------- /requirements/build.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements/mminstall.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/requirements/mminstall.txt -------------------------------------------------------------------------------- /requirements/optional.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements/readthedocs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/requirements/readthedocs.txt -------------------------------------------------------------------------------- /requirements/runtime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/requirements/runtime.txt -------------------------------------------------------------------------------- /requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/requirements/tests.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_rbbox_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tests/test_rbbox_iou.py -------------------------------------------------------------------------------- /tests/test_uav_bd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tests/test_uav_bd.py -------------------------------------------------------------------------------- /tests/test_yoloR_anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tests/test_yoloR_anchor_generator.py -------------------------------------------------------------------------------- /tools/data_preprocess/uav_bd_make_ids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tools/data_preprocess/uav_bd_make_ids.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/misc/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tools/misc/browse_dataset.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggefan/rotmmdet/HEAD/tools/train.py --------------------------------------------------------------------------------