模型压缩和剪枝--基于darknet权重文件移植

当知道如何训练自己的数据集后,如何将自己的模型优化并且速度更快,文件更小,精度更高呢,这时候除了调整超参数外,就要进行模型压缩和剪枝,并且为了落地到平台部署,推理速度是必须要考虑的,现在很多板卡(海思3519av100,jetson nano)上运行yolov3-tiny模型是可以跑的,FPS可以达到30左右,但是要检测多目标,或者复杂的物体识别,那就无法满足要求了。
因此我们要在保证准确度的前提下,对算法模型进行压缩和剪枝
参考开源项目:模型压缩和剪枝
https://github.com/tanluren/yolov3-channel-and-layer-pruning
而模型环境配置和demo示例遇到的一些问题可以参考这篇博客
博客
https://blog.youkuaiyun.com/avideointerfaces/article/details/103389902?utm_medium=distribute.pc_relevant.none-task-blog-baidujs-2
需要注意的是darknet一般是将标签文件和图片放在同一个文件夹中,标签并没有单独拿出来。且数据集的图片文件夹名字不能是images,
如果你命名为images,那么就要把标签单独拿出来,labels与images并行在一个目录中。
类似这样的问题

Couldn't open file: data1/labels/imgtest4_116.txt

注意

评论 2
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值