Python-对YOLOv3做模型剪枝

时间:2022-08-30 21:52:47
【文件属性】:
文件名称:Python-对YOLOv3做模型剪枝
文件大小:51KB
文件格式:ZIP
更新时间:2022-08-30 21:52:47
Python开发-机器学习 用 YOLOv3 模型在一个开源的人手检测数据集 oxford hand 上做人手检测,并在此基础上做模型剪枝。对于该数据集,对 YOLOv3 进行 channel pruning 之后,模型的参数量、模型大小减少 80% ,FLOPs 降低 70%,前向推断的速度可以达到原来的 200%,同时可以保持 mAP 基本不变。
【文件预览】:
YOLOv3-model-pruning-master
----train.py(8KB)
----utils()
--------utils.py(12KB)
--------datasets.py(6KB)
--------parse_config.py(1KB)
--------logger.py(1KB)
--------augmentations.py(2KB)
--------prune_utils.py(6KB)
----test.py(4KB)
----weights()
--------download_weights.sh(303B)
----models.py(17KB)
----config()
--------prune_yolov3-hand.cfg(8KB)
--------oxfordhand.data(81B)
--------create_hand_model.sh(8KB)
--------yolov3.cfg(8KB)
--------yolov3-hand.cfg(8KB)
--------yolov3-tiny.cfg(2KB)
----LICENSE(1KB)
----test_prune.py(6KB)
----README.md(4KB)
----data()
--------oxfordhand.names(6B)
--------train.txt(163KB)
--------valid.txt(26KB)
--------README.md(373B)
--------converter.py(4KB)
----debug_utils.py(627B)

网友评论