公开(公告)号：CN118918116A

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

申请号：CN202410793929.8

申请日：2024-06-19

Applicant: 北京交通大学

Inventor： 王志鹏 ,  孙一凡 ,  戴真 ,  袁亭亭 ,  勾中华 ,  贾利民 ,  秦勇 ,  耿毅轩

IPC: G06T7/10 ,  G06V10/774 ,  G06N3/0464 ,  G06N3/0455 ,  G06V10/80 ,  G06V10/764

Abstract: 本发明提供一种基于多尺度注意力机制的轨道板裂缝分割方法及系统，属于基于计算机视觉的图像分割技术领域，本发明该方法针对轨道板裂缝在无人机图像中像素占比小的问题，利用在大规模数据集上预训练好的神经网络作为编码器，提升模型对于裂缝细节信息的提取与表征，采用基于混合注意力机制的特征融合模块，使得模型可以排除背景冗余信息的干扰，专注于裂缝特征以提升轨道板裂缝分割精度，并引入Focal loss和Dice loss损失函数以解决裂缝分割任务中正负样本不平衡以及正样本难以学习的问题。