├── .github └── ISSUE_TEMPLATE │ ├── bug_report.yml │ ├── config.yml │ ├── feature_request.yml │ └── question.yml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── NOTICE ├── README.md ├── README_cn.md ├── assets ├── 701class_cmp_horizontal.png ├── CustomDatasetTutorial.md ├── DAMO-YOLO.pdf ├── applications │ ├── cigarette_detection.png │ ├── facemask_detection.png │ ├── head_detection.png │ ├── helmet_detection.png │ ├── human_detection.png │ ├── nflhelmet_detection.jpg │ ├── smartphone_detection.png │ └── trafficsign_detection.png ├── curve.png ├── dog.jpg ├── logo.png ├── nano_curve.png └── overview.gif ├── configs ├── damoyolo_tinynasL18_Nm.py ├── damoyolo_tinynasL18_Ns.py ├── damoyolo_tinynasL20_N.py ├── damoyolo_tinynasL20_Nl.py ├── damoyolo_tinynasL20_T.py ├── damoyolo_tinynasL25_S.py ├── damoyolo_tinynasL35_M.py └── damoyolo_tinynasL45_L.py ├── damo ├── __init__.py ├── apis │ ├── __init__.py │ ├── detector_inference.py │ ├── detector_inference_trt.py │ └── detector_trainer.py ├── augmentations │ ├── __init__.py │ ├── box_level_augs │ │ ├── __init__.py │ │ ├── box_level_augs.py │ │ ├── color_augs.py │ │ ├── gaussian_maps.py │ │ └── geometric_augs.py │ └── scale_aware_aug.py ├── base_models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── nas_backbones │ │ │ ├── tinynas_L20_k1kx.txt │ │ │ ├── tinynas_L20_k1kx_nano.txt │ │ │ ├── tinynas_L25_k1kx.txt │ │ │ ├── tinynas_L35_kxkx.txt │ │ │ ├── tinynas_L45_kxkx.txt │ │ │ ├── tinynas_nano_large.txt │ │ │ ├── tinynas_nano_middle.txt │ │ │ └── tinynas_nano_small.txt │ │ ├── tinynas_csp.py │ │ ├── tinynas_mob.py │ │ └── tinynas_res.py │ ├── core │ │ ├── atss_assigner.py │ │ ├── bbox_calculator.py │ │ ├── end2end.py │ │ ├── ops.py │ │ ├── ota_assigner.py │ │ ├── utils.py │ │ └── weight_init.py │ ├── heads │ │ ├── __init__.py │ │ └── zero_head.py │ ├── losses │ │ ├── distill_loss.py │ │ └── gfocal_loss.py │ └── necks │ │ ├── __init__.py │ │ └── giraffe_fpn_btn.py ├── config │ ├── __init__.py │ ├── augmentations.py │ ├── base.py │ └── paths_catalog.py ├── dataset │ ├── __init__.py │ ├── build.py │ ├── collate_batch.py │ ├── datasets │ │ ├── __init__.py │ │ ├── coco.py │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ └── coco │ │ │ │ ├── __init__.py │ │ │ │ └── coco_eval.py │ │ └── mosaic_wrapper.py │ ├── samplers │ │ ├── __init__.py │ │ ├── distributed.py │ │ ├── grouped_batch_sampler.py │ │ └── iteration_based_batch_sampler.py │ └── transforms │ │ ├── __init__.py │ │ ├── build.py │ │ ├── transforms.py │ │ ├── transforms_keepratio.py │ │ └── tta_aug.py ├── detectors │ └── detector.py ├── structures │ ├── __init__.py │ ├── bounding_box.py │ ├── boxlist_ops.py │ └── image_list.py └── utils │ ├── __init__.py │ ├── boxes.py │ ├── checkpoint.py │ ├── debug_utils.py │ ├── demo_utils.py │ ├── dist.py │ ├── imports.py │ ├── logger.py │ ├── metric.py │ ├── model_utils.py │ ├── timer.py │ └── visualize.py ├── datasets └── coco ├── requirements.txt ├── scripts ├── coco_distill.sh ├── coco_eval.sh └── coco_train.sh ├── setup.py └── tools ├── calibrator.py ├── converter.py ├── demo.py ├── eval.py ├── partial_quantization ├── README.md ├── partial_quant.py └── utils.py ├── train.py └── trt_eval.py /.github/ISSUE_TEMPLATE/bug_report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/.github/ISSUE_TEMPLATE/bug_report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- 1 | blank_issues_enabled: false 2 | -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/.github/ISSUE_TEMPLATE/feature_request.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/question.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/.github/ISSUE_TEMPLATE/question.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/README.md -------------------------------------------------------------------------------- /README_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/README_cn.md -------------------------------------------------------------------------------- /assets/701class_cmp_horizontal.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/701class_cmp_horizontal.png -------------------------------------------------------------------------------- /assets/CustomDatasetTutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/CustomDatasetTutorial.md -------------------------------------------------------------------------------- /assets/DAMO-YOLO.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/DAMO-YOLO.pdf -------------------------------------------------------------------------------- /assets/applications/cigarette_detection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/applications/cigarette_detection.png -------------------------------------------------------------------------------- /assets/applications/facemask_detection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/applications/facemask_detection.png -------------------------------------------------------------------------------- /assets/applications/head_detection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/applications/head_detection.png -------------------------------------------------------------------------------- /assets/applications/helmet_detection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/applications/helmet_detection.png -------------------------------------------------------------------------------- /assets/applications/human_detection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/applications/human_detection.png -------------------------------------------------------------------------------- /assets/applications/nflhelmet_detection.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/applications/nflhelmet_detection.jpg -------------------------------------------------------------------------------- /assets/applications/smartphone_detection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/applications/smartphone_detection.png -------------------------------------------------------------------------------- /assets/applications/trafficsign_detection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/applications/trafficsign_detection.png -------------------------------------------------------------------------------- /assets/curve.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/curve.png -------------------------------------------------------------------------------- /assets/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/dog.jpg -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/logo.png -------------------------------------------------------------------------------- /assets/nano_curve.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/nano_curve.png -------------------------------------------------------------------------------- /assets/overview.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/assets/overview.gif -------------------------------------------------------------------------------- /configs/damoyolo_tinynasL18_Nm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/configs/damoyolo_tinynasL18_Nm.py -------------------------------------------------------------------------------- /configs/damoyolo_tinynasL18_Ns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/configs/damoyolo_tinynasL18_Ns.py -------------------------------------------------------------------------------- /configs/damoyolo_tinynasL20_N.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/configs/damoyolo_tinynasL20_N.py -------------------------------------------------------------------------------- /configs/damoyolo_tinynasL20_Nl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/configs/damoyolo_tinynasL20_Nl.py -------------------------------------------------------------------------------- /configs/damoyolo_tinynasL20_T.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/configs/damoyolo_tinynasL20_T.py -------------------------------------------------------------------------------- /configs/damoyolo_tinynasL25_S.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/configs/damoyolo_tinynasL25_S.py -------------------------------------------------------------------------------- /configs/damoyolo_tinynasL35_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/configs/damoyolo_tinynasL35_M.py -------------------------------------------------------------------------------- /configs/damoyolo_tinynasL45_L.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/configs/damoyolo_tinynasL45_L.py -------------------------------------------------------------------------------- /damo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/__init__.py -------------------------------------------------------------------------------- /damo/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/apis/__init__.py -------------------------------------------------------------------------------- /damo/apis/detector_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/apis/detector_inference.py -------------------------------------------------------------------------------- /damo/apis/detector_inference_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/apis/detector_inference_trt.py -------------------------------------------------------------------------------- /damo/apis/detector_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/apis/detector_trainer.py -------------------------------------------------------------------------------- /damo/augmentations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/augmentations/__init__.py -------------------------------------------------------------------------------- /damo/augmentations/box_level_augs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/augmentations/box_level_augs/__init__.py -------------------------------------------------------------------------------- /damo/augmentations/box_level_augs/box_level_augs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/augmentations/box_level_augs/box_level_augs.py -------------------------------------------------------------------------------- /damo/augmentations/box_level_augs/color_augs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/augmentations/box_level_augs/color_augs.py -------------------------------------------------------------------------------- /damo/augmentations/box_level_augs/gaussian_maps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/augmentations/box_level_augs/gaussian_maps.py -------------------------------------------------------------------------------- /damo/augmentations/box_level_augs/geometric_augs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/augmentations/box_level_augs/geometric_augs.py -------------------------------------------------------------------------------- /damo/augmentations/scale_aware_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/augmentations/scale_aware_aug.py -------------------------------------------------------------------------------- /damo/base_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/__init__.py -------------------------------------------------------------------------------- /damo/base_models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/__init__.py -------------------------------------------------------------------------------- /damo/base_models/backbones/nas_backbones/tinynas_L20_k1kx.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/nas_backbones/tinynas_L20_k1kx.txt -------------------------------------------------------------------------------- /damo/base_models/backbones/nas_backbones/tinynas_L20_k1kx_nano.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/nas_backbones/tinynas_L20_k1kx_nano.txt -------------------------------------------------------------------------------- /damo/base_models/backbones/nas_backbones/tinynas_L25_k1kx.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/nas_backbones/tinynas_L25_k1kx.txt -------------------------------------------------------------------------------- /damo/base_models/backbones/nas_backbones/tinynas_L35_kxkx.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/nas_backbones/tinynas_L35_kxkx.txt -------------------------------------------------------------------------------- /damo/base_models/backbones/nas_backbones/tinynas_L45_kxkx.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/nas_backbones/tinynas_L45_kxkx.txt -------------------------------------------------------------------------------- /damo/base_models/backbones/nas_backbones/tinynas_nano_large.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/nas_backbones/tinynas_nano_large.txt -------------------------------------------------------------------------------- /damo/base_models/backbones/nas_backbones/tinynas_nano_middle.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/nas_backbones/tinynas_nano_middle.txt -------------------------------------------------------------------------------- /damo/base_models/backbones/nas_backbones/tinynas_nano_small.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/nas_backbones/tinynas_nano_small.txt -------------------------------------------------------------------------------- /damo/base_models/backbones/tinynas_csp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/tinynas_csp.py -------------------------------------------------------------------------------- /damo/base_models/backbones/tinynas_mob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/tinynas_mob.py -------------------------------------------------------------------------------- /damo/base_models/backbones/tinynas_res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/backbones/tinynas_res.py -------------------------------------------------------------------------------- /damo/base_models/core/atss_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/core/atss_assigner.py -------------------------------------------------------------------------------- /damo/base_models/core/bbox_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/core/bbox_calculator.py -------------------------------------------------------------------------------- /damo/base_models/core/end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/core/end2end.py -------------------------------------------------------------------------------- /damo/base_models/core/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/core/ops.py -------------------------------------------------------------------------------- /damo/base_models/core/ota_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/core/ota_assigner.py -------------------------------------------------------------------------------- /damo/base_models/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/core/utils.py -------------------------------------------------------------------------------- /damo/base_models/core/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/core/weight_init.py -------------------------------------------------------------------------------- /damo/base_models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/heads/__init__.py -------------------------------------------------------------------------------- /damo/base_models/heads/zero_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/heads/zero_head.py -------------------------------------------------------------------------------- /damo/base_models/losses/distill_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/losses/distill_loss.py -------------------------------------------------------------------------------- /damo/base_models/losses/gfocal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/losses/gfocal_loss.py -------------------------------------------------------------------------------- /damo/base_models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/necks/__init__.py -------------------------------------------------------------------------------- /damo/base_models/necks/giraffe_fpn_btn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/base_models/necks/giraffe_fpn_btn.py -------------------------------------------------------------------------------- /damo/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/config/__init__.py -------------------------------------------------------------------------------- /damo/config/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/config/augmentations.py -------------------------------------------------------------------------------- /damo/config/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/config/base.py -------------------------------------------------------------------------------- /damo/config/paths_catalog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/config/paths_catalog.py -------------------------------------------------------------------------------- /damo/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/__init__.py -------------------------------------------------------------------------------- /damo/dataset/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/build.py -------------------------------------------------------------------------------- /damo/dataset/collate_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/collate_batch.py -------------------------------------------------------------------------------- /damo/dataset/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/datasets/__init__.py -------------------------------------------------------------------------------- /damo/dataset/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/datasets/coco.py -------------------------------------------------------------------------------- /damo/dataset/datasets/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/datasets/evaluation/__init__.py -------------------------------------------------------------------------------- /damo/dataset/datasets/evaluation/coco/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/datasets/evaluation/coco/__init__.py -------------------------------------------------------------------------------- /damo/dataset/datasets/evaluation/coco/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/datasets/evaluation/coco/coco_eval.py -------------------------------------------------------------------------------- /damo/dataset/datasets/mosaic_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/datasets/mosaic_wrapper.py -------------------------------------------------------------------------------- /damo/dataset/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/samplers/__init__.py -------------------------------------------------------------------------------- /damo/dataset/samplers/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/samplers/distributed.py -------------------------------------------------------------------------------- /damo/dataset/samplers/grouped_batch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/samplers/grouped_batch_sampler.py -------------------------------------------------------------------------------- /damo/dataset/samplers/iteration_based_batch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/samplers/iteration_based_batch_sampler.py -------------------------------------------------------------------------------- /damo/dataset/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/transforms/__init__.py -------------------------------------------------------------------------------- /damo/dataset/transforms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/transforms/build.py -------------------------------------------------------------------------------- /damo/dataset/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/transforms/transforms.py -------------------------------------------------------------------------------- /damo/dataset/transforms/transforms_keepratio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/transforms/transforms_keepratio.py -------------------------------------------------------------------------------- /damo/dataset/transforms/tta_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/dataset/transforms/tta_aug.py -------------------------------------------------------------------------------- /damo/detectors/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/detectors/detector.py -------------------------------------------------------------------------------- /damo/structures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /damo/structures/bounding_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/structures/bounding_box.py -------------------------------------------------------------------------------- /damo/structures/boxlist_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/structures/boxlist_ops.py -------------------------------------------------------------------------------- /damo/structures/image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/structures/image_list.py -------------------------------------------------------------------------------- /damo/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/__init__.py -------------------------------------------------------------------------------- /damo/utils/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/boxes.py -------------------------------------------------------------------------------- /damo/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/checkpoint.py -------------------------------------------------------------------------------- /damo/utils/debug_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/debug_utils.py -------------------------------------------------------------------------------- /damo/utils/demo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/demo_utils.py -------------------------------------------------------------------------------- /damo/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/dist.py -------------------------------------------------------------------------------- /damo/utils/imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/imports.py -------------------------------------------------------------------------------- /damo/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/logger.py -------------------------------------------------------------------------------- /damo/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/metric.py -------------------------------------------------------------------------------- /damo/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/model_utils.py -------------------------------------------------------------------------------- /damo/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/timer.py -------------------------------------------------------------------------------- /damo/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/damo/utils/visualize.py -------------------------------------------------------------------------------- /datasets/coco: -------------------------------------------------------------------------------- 1 | /tmp/coco/coco_2017 -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/coco_distill.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/scripts/coco_distill.sh -------------------------------------------------------------------------------- /scripts/coco_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/scripts/coco_eval.sh -------------------------------------------------------------------------------- /scripts/coco_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/scripts/coco_train.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/setup.py -------------------------------------------------------------------------------- /tools/calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/calibrator.py -------------------------------------------------------------------------------- /tools/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/converter.py -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/demo.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/partial_quantization/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/partial_quantization/README.md -------------------------------------------------------------------------------- /tools/partial_quantization/partial_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/partial_quantization/partial_quant.py -------------------------------------------------------------------------------- /tools/partial_quantization/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/partial_quantization/utils.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/trt_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tinyvision/DAMO-YOLO/HEAD/tools/trt_eval.py --------------------------------------------------------------------------------