YOLOV5 RKNN部署加速

线程池推理

略(待补充)

修改导出模型的尺寸

python export.py --weights /workspace/yolov5/runs/train/exp12/weights/best.pt --img 640 --batch 1 --include onnx --opset 12

修改img的参数,要给根据自己的实际情况进行修改

修改检测头

略(待补充)

修改激活函数

例如seLU改为reLU,在性能上有所提升在精度上有所下降
对比如下
reLu精度
image
seLu精度
image
时间上快%25
修改过程 修改commony.py文件

default_act = nn.SiLU() # default activation # default_act = nn.ReLU() #default activation
替换。

注意: 更改激活函数后需要从头进行训练

posted @ 2025-12-03 18:11  shydragon  阅读(0)  评论(0)    收藏  举报