010-61934861

举报
帖子标题
优化yolov5模型在芯片上的推理性能小tips
*举报原因
取消
首页 > 开发者论坛 > 优化yolov5模型在芯片上的推理性能小tips
优化yolov5模型在芯片上的推理性能小tips
Flower 2024-01-04 17:12
举报 212 1 1

在实际部署中,往往都很关注推理性能,如果将yolov5模型部署在清微芯片上,有哪些方法可以提升板端推理性能?

方法一

     将yolo模型中的激活函数sigmoid等改为relu,可带来性能的明显提升;

方法二

     把后处理的sigmoid放在模型中,也可进一步优化

企业微信截图_1702884487130.png

发表评论
全部评论(1)
59 2024-03-02 20:59
回复
如果输入是定点,不管输出是什么数据类型,可以采用查表方式实现sigmoid。该方法不仅适合指令编程,也适合搞硬件加速器。
取消