【Pruning系列:二】Learning Efficient Convolutional Networks through Network Slimming|YOLOv3实践 |Pytorch 总结

本文介绍了YOLOv3模型的剪枝实践,包括基于Network Slimming的channel pruning和layer pruning,通过剪枝实现模型瘦身80%、速度提升100%,同时保持精度。介绍了不同剪枝策略的特点,如Residual结构的优化,并提供了相关项目的GitHub链接供参考。此外,还提及了适用于实时UAV应用的SlimYOLOv3模型。
摘要由CSDN通过智能技术生成

YOLOv3-model-pruning

https://github.com/Lam1360/YOLOv3-model-pruning

YOLOv3模型剪枝,瘦身80%,提速100%,精度基本不变

Introduction

剪枝算法步骤

以下只是算法的大概步骤,具体实现过程中还要做 s 参数的尝试或者需要进行迭代式剪枝等。

  • 进行稀疏化训练
  • 基于 test_prune.py 文件进行剪枝,得到剪枝后的模型
  • 对剪枝后的模型进行微调

Results

- 参数数量 模型体积 Flops 前向推断耗时(2070 TI) mAP
Baseline (416) 61.5M 246.4MB 32.8B 15.0 ms 0.7692
  • 1
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值