公开(公告)号：CN118887433A

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

申请号：CN202410858360.9

申请日：2024-06-28

申请人： 南京航空航天大学

发明人： 薛金凯 ,  汪雪婷 ,  陆轶 ,  康鸣翠 ,  黄晨昱 ,  刘宇轩 ,  严锦添 ,  崔节轩

IPC分类号： G06V10/762 ,  G06V10/80 ,  G06V10/32 ,  G06V10/82 ,  G06N3/08

摘要： 本发明提供了一种基于改进SSD模型的实时性机坪目标检测和识别方法。通过将预处理后的数据集输入改进的SSD模型进行训练，得到训练好的模型。改进的SSD模型采用RFB‑Net300作为骨干网络，加入了batch‑normalization层，使用sub‑pixel convolution layer代替上采样，并通过MFPN模型融合低层和高层特征信息。通过引入CIoU损失函数改进了原有的损失函数。生成预测框的过程中，通过K‑means聚类算法优化先验框的大小，并计算目标物体间的距离。MFPN模型通过构建特征金字塔，实现多尺度特征融合，增强了对不同尺寸目标的检测能力。CIoU损失函数综合评估预测框与实际框之间的相似度，考虑了重叠区域、中心点距离、宽高比和尺度。通过上述改进，本发明显著提升了机坪目标检测和识别的实时性和准确性。