logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

yolo处理数据的实用代码

【代码】yolo处理数据的实用代码。

文章图片
torch-pruning剪枝yolov5

来决定是否使用训练好的最佳模型来剪枝,并进行微调。先进行模型训练,通过参数。

文章图片
#剪枝
yolov5部署到rk3399pro

pt转onnx的时候可带可不带sigmoid,带了的话在rknn推理的时候就不用写了,没带的话在推理的时候就要写,可以用netron对比一下带不带的模型结构区别。yolo.py里的Detect类对forward进行修改。rknn推理(去掉使用sigmoid的地方)onnx转为rknn的节点位置可能需要更改。上面给的的是不带的,下面是带的(输出节点位置根据自己的调整。onnx输出节点为3个。

文章图片
知识蒸馏以及yolov5的蒸馏

之前的模型轻量化说过了量化和剪枝,现在到蒸馏了。

文章图片
到底了