公开(公告)号：CN118446987A

公开(公告)日：2024-08-06

申请号：CN202410560733.4

申请日：2024-05-08

Applicant: 河北工业大学

Inventor： 杨泽青 ,  许康妮 ,  赵丽滨 ,  胡宁 ,  吴江鹏 ,  马鑫盟

IPC: G06T7/00 ,  G06T7/11 ,  G06T5/40 ,  G06V10/25 ,  G06V10/762 ,  G06V10/774 ,  G06V10/82 ,  G06N3/045 ,  G06N3/0464 ,  G06N3/084 ,  G06N3/088

Abstract: 本发明公开一种用于狭长密闭空间的舱段内表面腐蚀视觉检测方法，该方法首先对训练图像采用了图像中心裁剪、图像预处理，其中，图像预处理包括基于梯度联合的直方图灰度均衡化、无监督图像扩充，为模型训练提供质量较好的训练样本；之后采用基于改进YOLOv5s模型的检测模型，首先使用改进的K‑means算法重新聚类anchors，生成满足飞机舱段内表面腐蚀数据集特征的先验框；并将YOLOv5s neck端靠近Upsample结构的两个Conv模块替换为GSConv模块，提升模型的轻量化水平；引入Triplet注意力机制，提高模型特征利用率、泛化能力和鲁棒性；将CIoU损失函数替换为Inner‑SIoU损失函数降低模型过拟合风险，从而提高模型检测的准确性，改善模型对腐蚀误检、漏检问题。