├── Dockerfile ├── LICENSE ├── README.md ├── ckpt └── yolov5s.pt ├── data ├── coco.yaml ├── coco128.yaml ├── get_coco2017.sh └── voc.yaml ├── detect.py ├── hubconf.py ├── inference ├── images │ ├── bus.jpg │ └── zidane.jpg └── output │ ├── bus.jpg │ ├── coco_1.jpg │ └── zidane.jpg ├── models ├── __init__.py ├── common.py ├── experimental.py ├── mobile-yolo3_voc.yaml ├── mobile-yolo5l_voc.yaml ├── mobile-yolo5s.yaml ├── mobile-yolo5s_voc.yaml ├── onnx_export.py ├── yolo.py ├── yolov3-spp.yaml ├── yolov5l.yaml ├── yolov5l_voc.yaml ├── yolov5m.yaml ├── yolov5s.yaml ├── yolov5s_voc.yaml ├── yolov5x.yaml └── yolov5x_voc.yaml ├── outputs └── voc │ ├── label.png │ ├── results.txt │ ├── train_batch0.jpg │ ├── train_batch1.jpg │ └── train_batch2.jpg ├── pic ├── after_pruning_prob_05.jpg ├── after_pruning_thres_001.jpg ├── demo1.jpg ├── demo2.jpg ├── demo3.jpg ├── demo4.jpg ├── distillation_1.jpg ├── distillation_2.png ├── distillation_3.png ├── distillation_4.jpg ├── distillation_5.png ├── distillation_6.jpg ├── distillation_7.png ├── pruning_func.jpg └── pruning_func.png ├── pruning.py ├── requirements.txt ├── script ├── cal_flop.py ├── coco2darknet.py ├── kmeans.py ├── labelme2darknet.py ├── split_train_data.py ├── tfyolo2darknet.py └── voc_label.py ├── test.py ├── torch_pruning ├── __init__.py ├── dependency.py ├── prune │ ├── __init__.py │ ├── structured.py │ └── unstructured.py └── utils.py ├── train.py ├── tutorial.ipynb ├── utils ├── __init__.py ├── activations.py ├── datasets.py ├── google_utils.py ├── torch_utils.py └── utils.py └── weights └── download_weights.sh /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/README.md -------------------------------------------------------------------------------- /ckpt/yolov5s.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/ckpt/yolov5s.pt -------------------------------------------------------------------------------- /data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/data/coco.yaml -------------------------------------------------------------------------------- /data/coco128.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/data/coco128.yaml -------------------------------------------------------------------------------- /data/get_coco2017.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/data/get_coco2017.sh -------------------------------------------------------------------------------- /data/voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/data/voc.yaml -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/detect.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/hubconf.py -------------------------------------------------------------------------------- /inference/images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/inference/images/bus.jpg -------------------------------------------------------------------------------- /inference/images/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/inference/images/zidane.jpg -------------------------------------------------------------------------------- /inference/output/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/inference/output/bus.jpg -------------------------------------------------------------------------------- /inference/output/coco_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/inference/output/coco_1.jpg -------------------------------------------------------------------------------- /inference/output/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/inference/output/zidane.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/mobile-yolo3_voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/mobile-yolo3_voc.yaml -------------------------------------------------------------------------------- /models/mobile-yolo5l_voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/mobile-yolo5l_voc.yaml -------------------------------------------------------------------------------- /models/mobile-yolo5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/mobile-yolo5s.yaml -------------------------------------------------------------------------------- /models/mobile-yolo5s_voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/mobile-yolo5s_voc.yaml -------------------------------------------------------------------------------- /models/onnx_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/onnx_export.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolov5l.yaml -------------------------------------------------------------------------------- /models/yolov5l_voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolov5l_voc.yaml -------------------------------------------------------------------------------- /models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolov5m.yaml -------------------------------------------------------------------------------- /models/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolov5s.yaml -------------------------------------------------------------------------------- /models/yolov5s_voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolov5s_voc.yaml -------------------------------------------------------------------------------- /models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolov5x.yaml -------------------------------------------------------------------------------- /models/yolov5x_voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/models/yolov5x_voc.yaml -------------------------------------------------------------------------------- /outputs/voc/label.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/outputs/voc/label.png -------------------------------------------------------------------------------- /outputs/voc/results.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/outputs/voc/results.txt -------------------------------------------------------------------------------- /outputs/voc/train_batch0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/outputs/voc/train_batch0.jpg -------------------------------------------------------------------------------- /outputs/voc/train_batch1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/outputs/voc/train_batch1.jpg -------------------------------------------------------------------------------- /outputs/voc/train_batch2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/outputs/voc/train_batch2.jpg -------------------------------------------------------------------------------- /pic/after_pruning_prob_05.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/after_pruning_prob_05.jpg -------------------------------------------------------------------------------- /pic/after_pruning_thres_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/after_pruning_thres_001.jpg -------------------------------------------------------------------------------- /pic/demo1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/demo1.jpg -------------------------------------------------------------------------------- /pic/demo2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/demo2.jpg -------------------------------------------------------------------------------- /pic/demo3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/demo3.jpg -------------------------------------------------------------------------------- /pic/demo4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/demo4.jpg -------------------------------------------------------------------------------- /pic/distillation_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/distillation_1.jpg -------------------------------------------------------------------------------- /pic/distillation_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/distillation_2.png -------------------------------------------------------------------------------- /pic/distillation_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/distillation_3.png -------------------------------------------------------------------------------- /pic/distillation_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/distillation_4.jpg -------------------------------------------------------------------------------- /pic/distillation_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/distillation_5.png -------------------------------------------------------------------------------- /pic/distillation_6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/distillation_6.jpg -------------------------------------------------------------------------------- /pic/distillation_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/distillation_7.png -------------------------------------------------------------------------------- /pic/pruning_func.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/pruning_func.jpg -------------------------------------------------------------------------------- /pic/pruning_func.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pic/pruning_func.png -------------------------------------------------------------------------------- /pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/pruning.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.0 -------------------------------------------------------------------------------- /script/cal_flop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/script/cal_flop.py -------------------------------------------------------------------------------- /script/coco2darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/script/coco2darknet.py -------------------------------------------------------------------------------- /script/kmeans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/script/kmeans.py -------------------------------------------------------------------------------- /script/labelme2darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/script/labelme2darknet.py -------------------------------------------------------------------------------- /script/split_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/script/split_train_data.py -------------------------------------------------------------------------------- /script/tfyolo2darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/script/tfyolo2darknet.py -------------------------------------------------------------------------------- /script/voc_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/script/voc_label.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/test.py -------------------------------------------------------------------------------- /torch_pruning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/torch_pruning/__init__.py -------------------------------------------------------------------------------- /torch_pruning/dependency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/torch_pruning/dependency.py -------------------------------------------------------------------------------- /torch_pruning/prune/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/torch_pruning/prune/__init__.py -------------------------------------------------------------------------------- /torch_pruning/prune/structured.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/torch_pruning/prune/structured.py -------------------------------------------------------------------------------- /torch_pruning/prune/unstructured.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/torch_pruning/prune/unstructured.py -------------------------------------------------------------------------------- /torch_pruning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/torch_pruning/utils.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/train.py -------------------------------------------------------------------------------- /tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/tutorial.ipynb -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/utils/utils.py -------------------------------------------------------------------------------- /weights/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syencil/mobile-yolov5-pruning-distillation/HEAD/weights/download_weights.sh --------------------------------------------------------------------------------