Python-对YOLOv3做模型剪枝

上传者: 39841856 | 上传时间: 2021-08-24 20:20:48 | 文件大小: 51KB | 文件类型: ZIP
用 YOLOv3 模型在一个开源的人手检测数据集 oxford hand 上做人手检测,并在此基础上做模型剪枝。对于该数据集,对 YOLOv3 进行 channel pruning 之后,模型的参数量、模型大小减少 80% ,FLOPs 降低 70%,前向推断的速度可以达到原来的 200%,同时可以保持 mAP 基本不变。

文件下载

资源详情

[{"title":"( 25 个子文件 51KB ) Python-对YOLOv3做模型剪枝","children":[{"title":"YOLOv3-model-pruning-master","children":[{"title":"train.py <span style='color:#111;'> 7.89KB </span>","children":null,"spread":false},{"title":"utils","children":[{"title":"utils.py <span style='color:#111;'> 11.76KB </span>","children":null,"spread":false},{"title":"datasets.py <span style='color:#111;'> 6.23KB </span>","children":null,"spread":false},{"title":"parse_config.py <span style='color:#111;'> 1.24KB </span>","children":null,"spread":false},{"title":"logger.py <span style='color:#111;'> 1.24KB </span>","children":null,"spread":false},{"title":"augmentations.py <span style='color:#111;'> 1.77KB </span>","children":null,"spread":false},{"title":"prune_utils.py <span style='color:#111;'> 6.27KB </span>","children":null,"spread":false}],"spread":true},{"title":"test.py <span style='color:#111;'> 3.99KB </span>","children":null,"spread":false},{"title":"weights","children":[{"title":"download_weights.sh <span style='color:#111;'> 303B </span>","children":null,"spread":false}],"spread":true},{"title":"models.py <span style='color:#111;'> 16.51KB </span>","children":null,"spread":false},{"title":"config","children":[{"title":"prune_yolov3-hand.cfg <span style='color:#111;'> 7.96KB </span>","children":null,"spread":false},{"title":"oxfordhand.data <span style='color:#111;'> 81B </span>","children":null,"spread":false},{"title":"create_hand_model.sh <span style='color:#111;'> 8.33KB </span>","children":null,"spread":false},{"title":"yolov3.cfg <span style='color:#111;'> 8.21KB </span>","children":null,"spread":false},{"title":"yolov3-hand.cfg <span style='color:#111;'> 8.14KB </span>","children":null,"spread":false},{"title":"yolov3-tiny.cfg <span style='color:#111;'> 1.98KB </span>","children":null,"spread":false}],"spread":true},{"title":"LICENSE <span style='color:#111;'> 1.04KB </span>","children":null,"spread":false},{"title":"test_prune.py <span style='color:#111;'> 5.89KB </span>","children":null,"spread":false},{"title":"README.md <span style='color:#111;'> 4.00KB </span>","children":null,"spread":false},{"title":"data","children":[{"title":"oxfordhand.names <span style='color:#111;'> 6B </span>","children":null,"spread":false},{"title":"train.txt <span style='color:#111;'> 163.00KB </span>","children":null,"spread":false},{"title":"valid.txt <span style='color:#111;'> 26.25KB </span>","children":null,"spread":false},{"title":"README.md <span style='color:#111;'> 373B </span>","children":null,"spread":false},{"title":"converter.py <span style='color:#111;'> 4.49KB </span>","children":null,"spread":false}],"spread":true},{"title":"debug_utils.py <span style='color:#111;'> 627B </span>","children":null,"spread":false}],"spread":false}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明