对YOLOv3做模型剪枝

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

相关的项目 - 更多比较

473 9.6k 3.5k

G gensim - Python库用于主题建模,文档索引和相似性检索大全集。目标受众是自然语言处理(NLP)和信息检索(IR)社区。
 
10.0 4.5
  10天前
345 9.4k 4.4k

Z 在视觉,文本,强化学习等方面围绕pytorch实现的一套例子
 
10.0 4.2
  6天前