├── .github └── FUNDING.yml ├── .gitignore ├── README.md ├── __init__.py ├── configs ├── coco.yaml ├── strongerv1.yaml ├── strongerv2.yaml ├── strongerv2_prune.yaml ├── strongerv2_sparse.yaml ├── strongerv3.yaml ├── strongerv3_0.75.yaml ├── strongerv3_1gt.yaml ├── strongerv3_all.yaml ├── strongerv3_asff.yaml ├── strongerv3_clsfocal.yaml ├── strongerv3_diou.yaml ├── strongerv3_kl.yaml ├── strongerv3_nosep.yaml ├── strongerv3_prune.yaml └── strongerv3_sparse.yaml ├── dataset ├── BaseDataset.py ├── __init__.py ├── augment │ ├── __init__.py │ ├── bbox.py │ ├── dataAug.py │ ├── image.py │ └── transform.py ├── coco.py ├── pascal.py └── pycocotools │ ├── _mask.c │ ├── _mask.cpython-36m-x86_64-linux-gnu.so │ ├── _mask.pyx │ ├── coco.py │ ├── cocoeval.py │ └── mask.py ├── docs ├── BNdistribute.png ├── MODELZOO.md └── Usage.md ├── evaluator ├── Evaluator.py ├── __init__.py ├── cocoeval.py └── voceval.py ├── main.py ├── main_prune.py ├── models ├── __init__.py ├── backbone │ ├── __init__.py │ ├── baseblock.py │ ├── darknet.py │ ├── helper.py │ └── mobilev2.py ├── strongerv1.py ├── strongerv3.py ├── strongerv3kl.py └── yololoss.py ├── pruning ├── BasePruner.py ├── Block.py ├── l1norm.py └── slimming.py ├── trainers ├── __init__.py ├── base_trainer.py ├── trainer_coco.py └── trainer_voc.py ├── utils ├── GIOU.py ├── dataset_util.py ├── get_kmeans.py ├── nms_utils.py ├── prune_utils.py ├── util.py └── visualize.py └── yacscfg.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # https://yihui-he.github.io/Stronger-yolo-pytorch/ 2 | -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/coco.yaml -------------------------------------------------------------------------------- /configs/strongerv1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv1.yaml -------------------------------------------------------------------------------- /configs/strongerv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv2.yaml -------------------------------------------------------------------------------- /configs/strongerv2_prune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv2_prune.yaml -------------------------------------------------------------------------------- /configs/strongerv2_sparse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv2_sparse.yaml -------------------------------------------------------------------------------- /configs/strongerv3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3.yaml -------------------------------------------------------------------------------- /configs/strongerv3_0.75.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_0.75.yaml -------------------------------------------------------------------------------- /configs/strongerv3_1gt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_1gt.yaml -------------------------------------------------------------------------------- /configs/strongerv3_all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_all.yaml -------------------------------------------------------------------------------- /configs/strongerv3_asff.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_asff.yaml -------------------------------------------------------------------------------- /configs/strongerv3_clsfocal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_clsfocal.yaml -------------------------------------------------------------------------------- /configs/strongerv3_diou.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_diou.yaml -------------------------------------------------------------------------------- /configs/strongerv3_kl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_kl.yaml -------------------------------------------------------------------------------- /configs/strongerv3_nosep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_nosep.yaml -------------------------------------------------------------------------------- /configs/strongerv3_prune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_prune.yaml -------------------------------------------------------------------------------- /configs/strongerv3_sparse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/configs/strongerv3_sparse.yaml -------------------------------------------------------------------------------- /dataset/BaseDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/BaseDataset.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/augment/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/augment/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/augment/bbox.py -------------------------------------------------------------------------------- /dataset/augment/dataAug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/augment/dataAug.py -------------------------------------------------------------------------------- /dataset/augment/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/augment/image.py -------------------------------------------------------------------------------- /dataset/augment/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/augment/transform.py -------------------------------------------------------------------------------- /dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/coco.py -------------------------------------------------------------------------------- /dataset/pascal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/pascal.py -------------------------------------------------------------------------------- /dataset/pycocotools/_mask.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/pycocotools/_mask.c -------------------------------------------------------------------------------- /dataset/pycocotools/_mask.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/pycocotools/_mask.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /dataset/pycocotools/_mask.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/pycocotools/_mask.pyx -------------------------------------------------------------------------------- /dataset/pycocotools/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/pycocotools/coco.py -------------------------------------------------------------------------------- /dataset/pycocotools/cocoeval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/pycocotools/cocoeval.py -------------------------------------------------------------------------------- /dataset/pycocotools/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/dataset/pycocotools/mask.py -------------------------------------------------------------------------------- /docs/BNdistribute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/docs/BNdistribute.png -------------------------------------------------------------------------------- /docs/MODELZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/docs/MODELZOO.md -------------------------------------------------------------------------------- /docs/Usage.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/docs/Usage.md -------------------------------------------------------------------------------- /evaluator/Evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/evaluator/Evaluator.py -------------------------------------------------------------------------------- /evaluator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluator/cocoeval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/evaluator/cocoeval.py -------------------------------------------------------------------------------- /evaluator/voceval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/evaluator/voceval.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/main.py -------------------------------------------------------------------------------- /main_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/main_prune.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/baseblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/backbone/baseblock.py -------------------------------------------------------------------------------- /models/backbone/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/backbone/darknet.py -------------------------------------------------------------------------------- /models/backbone/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/backbone/helper.py -------------------------------------------------------------------------------- /models/backbone/mobilev2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/backbone/mobilev2.py -------------------------------------------------------------------------------- /models/strongerv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/strongerv1.py -------------------------------------------------------------------------------- /models/strongerv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/strongerv3.py -------------------------------------------------------------------------------- /models/strongerv3kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/strongerv3kl.py -------------------------------------------------------------------------------- /models/yololoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/models/yololoss.py -------------------------------------------------------------------------------- /pruning/BasePruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/pruning/BasePruner.py -------------------------------------------------------------------------------- /pruning/Block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/pruning/Block.py -------------------------------------------------------------------------------- /pruning/l1norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/pruning/l1norm.py -------------------------------------------------------------------------------- /pruning/slimming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/pruning/slimming.py -------------------------------------------------------------------------------- /trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/trainers/__init__.py -------------------------------------------------------------------------------- /trainers/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/trainers/base_trainer.py -------------------------------------------------------------------------------- /trainers/trainer_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/trainers/trainer_coco.py -------------------------------------------------------------------------------- /trainers/trainer_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/trainers/trainer_voc.py -------------------------------------------------------------------------------- /utils/GIOU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/utils/GIOU.py -------------------------------------------------------------------------------- /utils/dataset_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/utils/dataset_util.py -------------------------------------------------------------------------------- /utils/get_kmeans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/utils/get_kmeans.py -------------------------------------------------------------------------------- /utils/nms_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/utils/nms_utils.py -------------------------------------------------------------------------------- /utils/prune_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/utils/prune_utils.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/utils/util.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/utils/visualize.py -------------------------------------------------------------------------------- /yacscfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/Stronger-yolo-pytorch/HEAD/yacscfg.py --------------------------------------------------------------------------------