1. 首页
  2. 移动开发
  3. 其他
  4. Python 对YOLOv3做模型剪枝

Python 对YOLOv3做模型剪枝

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