轻量水下检测网络

算力低也能实时跑

基于YOLOv5-lite与动态采样Transformer,以相同参数量显著提升水下目标检测精度。

陈亮 · Yang, Yuyi · Zhenheng, Wang · 张剑 · 周少武 · Wu, Lianghong

Journal of Marine Science and Engineering 2023

参数信息

检测精度提升
6.6 %
检测精度提升
5.0 %
计算效率提升
58.8 %

技术优势

参数不增,精度更高

动态采样 Transformer 模块仅通过稀疏采样和移位注意力实现针对性建模,未增加参数量,却让模型在参数量不变的前提下提升检测精度。

省内存,前向更轻

共享核参数卷积被用于优化矩阵编码,简化了前向传播的内存开销,使模型更适合部署在算力有限的水下机器人上。

定位知识传得更准

训练时采用解耦定位与识别的蒸馏方法,强化正样本框的有效定位知识传递,模型参数不变即可提升检测精度。

应用场景