这里介绍如何对YOLOv5网络进行修改,让其可以在边缘设备,例如RK3588上较好的运行。
由于默认YOLOv5的输出形状为[1,3,25200,85],对于边缘设备而言,此时需要将其拆分为3个输出比较妥当。
而优化后网络第17层如下:
而对应网络第23层如下:
可以看到,其中的SiLU被替换为ReLU,从而可以在边缘上较好的运行。另外原始网络中Detect模块的输出被拆分为3个输出,从而简化了其计算过程。
如果喜欢这篇文章或对您有帮助，可以：[☕] 请我喝杯咖啡 | [💓] 小额赞助
/ YOLOv5在边缘设备上的优化
Created Sat, 15 Nov 2025 10:22:02 +0800
Modified Sat, 15 Nov 2025 10:26:19 +0800