├── .gitignore ├── LICENSE ├── README.md ├── README_Orginal.md ├── data ├── Argoverse_HD.yaml ├── GlobalWheat2020.yaml ├── Objects365.yaml ├── SKU-110K.yaml ├── VOC.yaml ├── VisDrone.yaml ├── coco.yaml ├── coco128.yaml ├── fire.yaml ├── hyps │ ├── hyp.finetune.yaml │ ├── hyp.finetune_objects365.yaml │ ├── hyp.scratch-p6.yaml │ └── hyp.scratch.yaml ├── scripts │ ├── download_weights.sh │ ├── get_coco.sh │ └── get_coco128.sh └── xView.yaml ├── deep_sort.py ├── deep_sort ├── __init__.py ├── deep │ ├── .gitignore │ ├── LICENSE │ ├── README.md │ ├── __init__.py │ ├── checkpoint │ │ └── .gitkeep │ ├── eval.py │ ├── feature_extractor.py │ ├── model.py │ ├── models │ │ ├── __init__.py │ │ ├── densenet.py │ │ ├── hacnn.py │ │ ├── inceptionresnetv2.py │ │ ├── inceptionv4.py │ │ ├── mlfn.py │ │ ├── mobilenetv2.py │ │ ├── mudeep.py │ │ ├── nasnet.py │ │ ├── original_model.py │ │ ├── osnet.py │ │ ├── osnet_ain.py │ │ ├── pcb.py │ │ ├── resnet.py │ │ ├── resnet_ibn_a.py │ │ ├── resnet_ibn_b.py │ │ ├── resnetmid.py │ │ ├── senet.py │ │ ├── shufflenet.py │ │ ├── shufflenetv2.py │ │ ├── squeezenet.py │ │ └── xception.py │ ├── my_experiments │ │ └── train_mobilenetv2.sh │ ├── oldfeature_extractor.py │ ├── train.py │ ├── train_wo_center.py │ └── utils │ │ ├── assign_train_val.py │ │ ├── center_loss.py │ │ ├── compute_mean_std.py │ │ ├── pre_deep.py │ │ ├── rename_all.py │ │ ├── tsne_vis.py │ │ └── visualize_actmap.py ├── deep_sort.py └── sort │ ├── __init__.py │ ├── detection.py │ ├── iou_matching.py │ ├── kalman_filter.py │ ├── linear_assignment.py │ ├── nn_matching.py │ ├── preprocessing.py │ ├── track.py │ └── tracker.py ├── detect.py ├── detect_camera.py ├── export.py ├── get_small_model.py ├── get_sparse_model.py ├── hubconf.py ├── models ├── __init__.py ├── common.py ├── common_pruning.py ├── experimental.py ├── hub │ ├── anchors.yaml │ ├── yolov3-spp.yaml │ ├── yolov3-tiny.yaml │ ├── yolov3.yaml │ ├── yolov5-fpn.yaml │ ├── yolov5-p2.yaml │ ├── yolov5-p6.yaml │ ├── yolov5-p7.yaml │ ├── yolov5-panet.yaml │ ├── yolov5l6.yaml │ ├── yolov5m6.yaml │ ├── yolov5s-transformer.yaml │ ├── yolov5s6.yaml │ └── yolov5x6.yaml ├── yolo.py ├── yolov5_pruning.py ├── yolov5l.yaml ├── yolov5m.yaml ├── yolov5s.yaml └── yolov5x.yaml ├── my_experiments ├── detect_coco.sh ├── detect_fire.sh ├── detect_fire_prune.sh ├── export2onnx.sh ├── get_small_model.sh ├── get_sparse_model.sh ├── retrain_fire_pruning.sh ├── tensorboard.sh ├── train_coco128.sh ├── train_coco_network_slimming.sh ├── train_coco_prune.sh ├── train_fire.sh ├── train_fire_network_slimming.sh └── train_fire_prune.sh ├── requirements.txt ├── train.py ├── utils ├── __init__.py ├── activations.py ├── augmentations.py ├── autoanchor.py ├── aws │ ├── __init__.py │ ├── mime.sh │ ├── resume.py │ └── userdata.sh ├── checkpoints_convert.py ├── datasets.py ├── flask_rest_api │ ├── README.md │ ├── example_request.py │ └── restapi.py ├── general.py ├── google_app_engine │ ├── Dockerfile │ ├── additional_requirements.txt │ └── app.yaml ├── google_utils.py ├── loss.py ├── metrics.py ├── plots.py ├── prune_utils.py ├── torch_utils.py └── wandb_logging │ ├── __init__.py │ ├── log_dataset.py │ ├── sweep.py │ ├── sweep.yaml │ └── wandb_utils.py └── val.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/README.md -------------------------------------------------------------------------------- /README_Orginal.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/README_Orginal.md -------------------------------------------------------------------------------- /data/Argoverse_HD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/Argoverse_HD.yaml -------------------------------------------------------------------------------- /data/GlobalWheat2020.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/GlobalWheat2020.yaml -------------------------------------------------------------------------------- /data/Objects365.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/Objects365.yaml -------------------------------------------------------------------------------- /data/SKU-110K.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/SKU-110K.yaml -------------------------------------------------------------------------------- /data/VOC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/VOC.yaml -------------------------------------------------------------------------------- /data/VisDrone.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/VisDrone.yaml -------------------------------------------------------------------------------- /data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/coco.yaml -------------------------------------------------------------------------------- /data/coco128.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/coco128.yaml -------------------------------------------------------------------------------- /data/fire.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/fire.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/hyps/hyp.finetune.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.finetune_objects365.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/hyps/hyp.finetune_objects365.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.scratch-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/hyps/hyp.scratch-p6.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/hyps/hyp.scratch.yaml -------------------------------------------------------------------------------- /data/scripts/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/scripts/download_weights.sh -------------------------------------------------------------------------------- /data/scripts/get_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/scripts/get_coco.sh -------------------------------------------------------------------------------- /data/scripts/get_coco128.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/scripts/get_coco128.sh -------------------------------------------------------------------------------- /data/xView.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/data/xView.yaml -------------------------------------------------------------------------------- /deep_sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort.py -------------------------------------------------------------------------------- /deep_sort/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/__init__.py -------------------------------------------------------------------------------- /deep_sort/deep/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/.gitignore -------------------------------------------------------------------------------- /deep_sort/deep/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/LICENSE -------------------------------------------------------------------------------- /deep_sort/deep/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/README.md -------------------------------------------------------------------------------- /deep_sort/deep/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deep_sort/deep/checkpoint/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deep_sort/deep/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/eval.py -------------------------------------------------------------------------------- /deep_sort/deep/feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/feature_extractor.py -------------------------------------------------------------------------------- /deep_sort/deep/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/model.py -------------------------------------------------------------------------------- /deep_sort/deep/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/__init__.py -------------------------------------------------------------------------------- /deep_sort/deep/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/densenet.py -------------------------------------------------------------------------------- /deep_sort/deep/models/hacnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/hacnn.py -------------------------------------------------------------------------------- /deep_sort/deep/models/inceptionresnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/inceptionresnetv2.py -------------------------------------------------------------------------------- /deep_sort/deep/models/inceptionv4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/inceptionv4.py -------------------------------------------------------------------------------- /deep_sort/deep/models/mlfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/mlfn.py -------------------------------------------------------------------------------- /deep_sort/deep/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/mobilenetv2.py -------------------------------------------------------------------------------- /deep_sort/deep/models/mudeep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/mudeep.py -------------------------------------------------------------------------------- /deep_sort/deep/models/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/nasnet.py -------------------------------------------------------------------------------- /deep_sort/deep/models/original_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/original_model.py -------------------------------------------------------------------------------- /deep_sort/deep/models/osnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/osnet.py -------------------------------------------------------------------------------- /deep_sort/deep/models/osnet_ain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/osnet_ain.py -------------------------------------------------------------------------------- /deep_sort/deep/models/pcb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/pcb.py -------------------------------------------------------------------------------- /deep_sort/deep/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/resnet.py -------------------------------------------------------------------------------- /deep_sort/deep/models/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/resnet_ibn_a.py -------------------------------------------------------------------------------- /deep_sort/deep/models/resnet_ibn_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/resnet_ibn_b.py -------------------------------------------------------------------------------- /deep_sort/deep/models/resnetmid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/resnetmid.py -------------------------------------------------------------------------------- /deep_sort/deep/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/senet.py -------------------------------------------------------------------------------- /deep_sort/deep/models/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/shufflenet.py -------------------------------------------------------------------------------- /deep_sort/deep/models/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/shufflenetv2.py -------------------------------------------------------------------------------- /deep_sort/deep/models/squeezenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/squeezenet.py -------------------------------------------------------------------------------- /deep_sort/deep/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/models/xception.py -------------------------------------------------------------------------------- /deep_sort/deep/my_experiments/train_mobilenetv2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/my_experiments/train_mobilenetv2.sh -------------------------------------------------------------------------------- /deep_sort/deep/oldfeature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/oldfeature_extractor.py -------------------------------------------------------------------------------- /deep_sort/deep/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/train.py -------------------------------------------------------------------------------- /deep_sort/deep/train_wo_center.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/train_wo_center.py -------------------------------------------------------------------------------- /deep_sort/deep/utils/assign_train_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/utils/assign_train_val.py -------------------------------------------------------------------------------- /deep_sort/deep/utils/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/utils/center_loss.py -------------------------------------------------------------------------------- /deep_sort/deep/utils/compute_mean_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/utils/compute_mean_std.py -------------------------------------------------------------------------------- /deep_sort/deep/utils/pre_deep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/utils/pre_deep.py -------------------------------------------------------------------------------- /deep_sort/deep/utils/rename_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/utils/rename_all.py -------------------------------------------------------------------------------- /deep_sort/deep/utils/tsne_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/utils/tsne_vis.py -------------------------------------------------------------------------------- /deep_sort/deep/utils/visualize_actmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep/utils/visualize_actmap.py -------------------------------------------------------------------------------- /deep_sort/deep_sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/deep_sort.py -------------------------------------------------------------------------------- /deep_sort/sort/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deep_sort/sort/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/sort/detection.py -------------------------------------------------------------------------------- /deep_sort/sort/iou_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/sort/iou_matching.py -------------------------------------------------------------------------------- /deep_sort/sort/kalman_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/sort/kalman_filter.py -------------------------------------------------------------------------------- /deep_sort/sort/linear_assignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/sort/linear_assignment.py -------------------------------------------------------------------------------- /deep_sort/sort/nn_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/sort/nn_matching.py -------------------------------------------------------------------------------- /deep_sort/sort/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/sort/preprocessing.py -------------------------------------------------------------------------------- /deep_sort/sort/track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/sort/track.py -------------------------------------------------------------------------------- /deep_sort/sort/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/deep_sort/sort/tracker.py -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/detect.py -------------------------------------------------------------------------------- /detect_camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/detect_camera.py -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/export.py -------------------------------------------------------------------------------- /get_small_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/get_small_model.py -------------------------------------------------------------------------------- /get_sparse_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/get_sparse_model.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/hubconf.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/common.py -------------------------------------------------------------------------------- /models/common_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/common_pruning.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/hub/anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/anchors.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5l6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5m6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5s-transformer.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5s6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5x6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/hub/yolov5x6.yaml -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/yolov5_pruning.py -------------------------------------------------------------------------------- /models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/yolov5l.yaml -------------------------------------------------------------------------------- /models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/yolov5m.yaml -------------------------------------------------------------------------------- /models/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/yolov5s.yaml -------------------------------------------------------------------------------- /models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/models/yolov5x.yaml -------------------------------------------------------------------------------- /my_experiments/detect_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/detect_coco.sh -------------------------------------------------------------------------------- /my_experiments/detect_fire.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/detect_fire.sh -------------------------------------------------------------------------------- /my_experiments/detect_fire_prune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/detect_fire_prune.sh -------------------------------------------------------------------------------- /my_experiments/export2onnx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/export2onnx.sh -------------------------------------------------------------------------------- /my_experiments/get_small_model.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/get_small_model.sh -------------------------------------------------------------------------------- /my_experiments/get_sparse_model.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/get_sparse_model.sh -------------------------------------------------------------------------------- /my_experiments/retrain_fire_pruning.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/retrain_fire_pruning.sh -------------------------------------------------------------------------------- /my_experiments/tensorboard.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/tensorboard.sh -------------------------------------------------------------------------------- /my_experiments/train_coco128.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/train_coco128.sh -------------------------------------------------------------------------------- /my_experiments/train_coco_network_slimming.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/train_coco_network_slimming.sh -------------------------------------------------------------------------------- /my_experiments/train_coco_prune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/train_coco_prune.sh -------------------------------------------------------------------------------- /my_experiments/train_fire.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/train_fire.sh -------------------------------------------------------------------------------- /my_experiments/train_fire_network_slimming.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/train_fire_network_slimming.sh -------------------------------------------------------------------------------- /my_experiments/train_fire_prune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/my_experiments/train_fire_prune.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/aws/mime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/aws/mime.sh -------------------------------------------------------------------------------- /utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/aws/resume.py -------------------------------------------------------------------------------- /utils/aws/userdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/aws/userdata.sh -------------------------------------------------------------------------------- /utils/checkpoints_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/checkpoints_convert.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/flask_rest_api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/flask_rest_api/README.md -------------------------------------------------------------------------------- /utils/flask_rest_api/example_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/flask_rest_api/example_request.py -------------------------------------------------------------------------------- /utils/flask_rest_api/restapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/flask_rest_api/restapi.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/prune_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/prune_utils.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/wandb_logging/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/wandb_logging/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/wandb_logging/log_dataset.py -------------------------------------------------------------------------------- /utils/wandb_logging/sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/wandb_logging/sweep.py -------------------------------------------------------------------------------- /utils/wandb_logging/sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/wandb_logging/sweep.yaml -------------------------------------------------------------------------------- /utils/wandb_logging/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/utils/wandb_logging/wandb_utils.py -------------------------------------------------------------------------------- /val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DDGRCF/yolov5_pruning/HEAD/val.py --------------------------------------------------------------------------------