基于YOLOv3的布匹瑕疵检测方法

作者:伍洪健; 邓作杰*; 章银萍; 张金召; 王小康
来源:湖南工程学院学报(自然科学版), 2023, 33(03): 39-43.
DOI:10.15987/j.cnki.hgbjbz.2023.03.011

摘要

针对布匹瑕疵差异较大、分布不均匀等问题,在YOLOv3中引入SwinTransformerBlock模块,用自注意力机制专注于有效特征排除无效特征的干扰,解决瑕疵差异大、分布不均等问题.同时用可变形卷积v2替换普通卷积,增大网络的感受野和多尺度建模能力,更好地适应瑕疵的形状和位置变化,从而提高目标检测的准确性和鲁棒性.实验结果表明,改进后算法在mAP上比原算法提高了3.80%,在检测速度上下降了2.86帧每秒.

全文