公开(公告)号：CN116645586A

公开(公告)日：2023-08-25

申请号：CN202310626179.0

申请日：2023-05-30

Applicant: 无锡学院

Inventor： 裴晓芳 ,  刘菁宇

IPC: G06V10/82 ,  G06V20/52 ,  G06V20/40 ,  G06V10/52 ,  G06V10/80 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/048 ,  G06N3/084

Abstract: 本发明公开了一种基于改进YOLOv5的港口集装箱损伤检测方法及系统，所述方法包括以下步骤：对获取到的港口集装箱损伤图像进行预处理和标注，划分为训练集和测试集；改进YOLOv5模型：以YOLOv5模型为基本架构，构造损伤检测模型，在Backbone网络结构中最后一层之后加入CBAM注意力机制；Neck网络结构中加入BiFPN网络来进行目标特征的融合；设置损失函数EIOU Loss替代原YOLOv5网络中定位损失函数；利用训练集对损伤检测模型进行训练，直到所述损伤检测模型的目标损失函数收敛；利用训练好的损伤检测模型对测试集进行检测，输出港口集装箱损伤图像检测结果，计算平均精度AP和平均精度均值mAP。通过本发明方法，实现复杂港口环境下集装箱多种损伤类型的高精度检测。