一种基于深度学习的管线浅剖图像自动识别系统

本发明公开了一种基于深度学习的管线浅剖图像自动识别系统及方法，包含引入模块、模型改进模块、数据采集模块、训练模块和识别模块。模型改进模块在基础模型中融入注意力特征金字塔网络、目标检测动态头部结构和优化损失函数形成初始识别模型。注意力特征金字塔网络动态调整特征图提取权重；目标检测动态头部结构根据图像复杂度调整参数；优化损失函数调整角度、尺度和长宽比的惩罚项。数据采集模块采集形成训练集、测试集和验证集。训练模块基于训练集数据训练初始模型，通过验证集调整超参数直至模型性能满足预设条件。识别模块使用优化模型对测试集图像进行识别得到最终结果。本发明可有效提升管线浅剖图像的自动识别精度与效率。