├── .gitignore ├── LICENSE ├── README.md ├── configs ├── baseline.yaml └── mutual_guide.yaml ├── data ├── __init__.py ├── coco.py ├── data_augment.py ├── data_prefetcher.py ├── scripts │ ├── COCO2014.sh │ ├── COCO2017.sh │ ├── VOC2007.sh │ └── VOC2012.sh ├── voc0712.py ├── voc_eval.py └── xml_dataset.py ├── distil.py ├── doc ├── compare.png └── mg.svg ├── models ├── __init__.py ├── backbone │ ├── __init__.py │ ├── cspdarknet_backbone.py │ ├── efficientnet_backbone.py │ ├── efficientnetv2_backbone.py │ ├── gpunet_backbone.py │ ├── regnet_backbone.py │ ├── repvgg_backbone.py │ ├── resnet_backbone.py │ ├── shufflenet_backbone.py │ ├── swin_backbone.py │ └── vgg_backbone.py ├── base_blocks.py ├── detector.py └── neck │ ├── __init__.py │ ├── fpn_neck.py │ ├── pafpn_neck.py │ └── ssd_neck.py ├── test.py ├── train.py └── utils ├── __init__.py ├── box ├── __init__.py ├── box_utils.py ├── detection.py ├── prior_box.py └── seq_matcher.py ├── ema.py ├── flops_counter.py ├── loss ├── __init__.py ├── balanced_l1_loss.py ├── focal_loss.py ├── gfocal_loss.py ├── giou_loss.py ├── hint_loss.py ├── multibox_loss.py └── siou_loss.py ├── lr_scheduler.py └── timer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/README.md -------------------------------------------------------------------------------- /configs/baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/configs/baseline.yaml -------------------------------------------------------------------------------- /configs/mutual_guide.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/configs/mutual_guide.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/coco.py -------------------------------------------------------------------------------- /data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/data_augment.py -------------------------------------------------------------------------------- /data/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/data_prefetcher.py -------------------------------------------------------------------------------- /data/scripts/COCO2014.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/scripts/COCO2014.sh -------------------------------------------------------------------------------- /data/scripts/COCO2017.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/scripts/COCO2017.sh -------------------------------------------------------------------------------- /data/scripts/VOC2007.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/scripts/VOC2007.sh -------------------------------------------------------------------------------- /data/scripts/VOC2012.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/scripts/VOC2012.sh -------------------------------------------------------------------------------- /data/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/voc0712.py -------------------------------------------------------------------------------- /data/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/voc_eval.py -------------------------------------------------------------------------------- /data/xml_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/data/xml_dataset.py -------------------------------------------------------------------------------- /distil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/distil.py -------------------------------------------------------------------------------- /doc/compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/doc/compare.png -------------------------------------------------------------------------------- /doc/mg.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/doc/mg.svg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/cspdarknet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/cspdarknet_backbone.py -------------------------------------------------------------------------------- /models/backbone/efficientnet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/efficientnet_backbone.py -------------------------------------------------------------------------------- /models/backbone/efficientnetv2_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/efficientnetv2_backbone.py -------------------------------------------------------------------------------- /models/backbone/gpunet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/gpunet_backbone.py -------------------------------------------------------------------------------- /models/backbone/regnet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/regnet_backbone.py -------------------------------------------------------------------------------- /models/backbone/repvgg_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/repvgg_backbone.py -------------------------------------------------------------------------------- /models/backbone/resnet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/resnet_backbone.py -------------------------------------------------------------------------------- /models/backbone/shufflenet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/shufflenet_backbone.py -------------------------------------------------------------------------------- /models/backbone/swin_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/swin_backbone.py -------------------------------------------------------------------------------- /models/backbone/vgg_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/backbone/vgg_backbone.py -------------------------------------------------------------------------------- /models/base_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/base_blocks.py -------------------------------------------------------------------------------- /models/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/detector.py -------------------------------------------------------------------------------- /models/neck/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/neck/__init__.py -------------------------------------------------------------------------------- /models/neck/fpn_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/neck/fpn_neck.py -------------------------------------------------------------------------------- /models/neck/pafpn_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/neck/pafpn_neck.py -------------------------------------------------------------------------------- /models/neck/ssd_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/models/neck/ssd_neck.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/box/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/box/__init__.py -------------------------------------------------------------------------------- /utils/box/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/box/box_utils.py -------------------------------------------------------------------------------- /utils/box/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/box/detection.py -------------------------------------------------------------------------------- /utils/box/prior_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/box/prior_box.py -------------------------------------------------------------------------------- /utils/box/seq_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/box/seq_matcher.py -------------------------------------------------------------------------------- /utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/ema.py -------------------------------------------------------------------------------- /utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/flops_counter.py -------------------------------------------------------------------------------- /utils/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/loss/__init__.py -------------------------------------------------------------------------------- /utils/loss/balanced_l1_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/loss/balanced_l1_loss.py -------------------------------------------------------------------------------- /utils/loss/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/loss/focal_loss.py -------------------------------------------------------------------------------- /utils/loss/gfocal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/loss/gfocal_loss.py -------------------------------------------------------------------------------- /utils/loss/giou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/loss/giou_loss.py -------------------------------------------------------------------------------- /utils/loss/hint_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/loss/hint_loss.py -------------------------------------------------------------------------------- /utils/loss/multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/loss/multibox_loss.py -------------------------------------------------------------------------------- /utils/loss/siou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/loss/siou_loss.py -------------------------------------------------------------------------------- /utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/lr_scheduler.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghengdev/MutualGuide/HEAD/utils/timer.py --------------------------------------------------------------------------------