pytorch 48 基于voc2012数据集对yolov10s模型进行剪枝尝试

48 篇文章 200 订阅 ¥59.90 ¥99.00

对yolov10s模型进行剪枝,以voc2012数据集做研究对象(训练数据 voc2012 train; 测试数据 voc2012 val),移除50%的通道,28.5 GFLOPs降低到14.6 GFLOPs,理论速度提升1倍。剪枝后得到的模型的最高map50为71(相比于剪枝前的74,下降3个点),map5095最高为54(相比于剪枝前的58,下降4个点)。并基于trt-python进行fps的差异对比,发现原始模型的fp16 推理fps为380,剪枝后模型fps为405。同时,也基于onnx进行推理耗时测试,gpu下,原始模型推理耗时为16ms,剪枝模型耗时为13毫秒;cpu下,原始模型推理耗时为66ms,剪枝后模型推理耗时为49ms。

1、完整代码

1.1 依赖函数

在ultralytics项目根目录下创建user_models目录,并将以下代码保存为user_models下prune_model.py。
代码主要是实现C2f_v2类与C2fCIB_v2,用于在模型剪枝前替换掉yolov10中的C2f与C2fCIB模块


from ultraly
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

万里鹏程转瞬至

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值