参数缩减近七成
通过特征平衡与小型目标增强,在边缘设备上实现多尺度故障的高效检测。
韩谷静 · Ruijie, Wang · Qiwei, Yuan · Zhao, Liu · Saidian, Li · 张明 · He, Min · 秦亮
Drones 2023
用 Ghost 模块轻量化特征提取与预测网络,参数量比 YOLOv7-Tiny 减少 74.79%,更容易放进无人机机载设备。
通过特征平衡和小目标增强,mAP 比 YOLOv7-Tiny 还提高 0.71%,轻量化没有牺牲检测能力。
引入 NWD 损失并优化与 CIoU 的融合比例,专门补偿轻量化带来的小目标精度损失。
部署到 Jetson Xavier NX 模拟巡检时达到 23.5 FPS,在算力紧张的机载设备上也能实时推理。