基于YOLO v3-tiny的红外遥感图像跑道检测方法
摘要:
本发明涉及一种基于YOLO v3‑tiny的红外遥感图像跑道检测方法,包括下列步骤:构建红外遥感机场跑道图像数据集;确定训练集和验证集,对训练集和验证集进行标注;建立YOLO v3‑tiny检测模型;网络参数设置;训练YOLO v3‑tiny检测模型;基于线段检测器的预训练模型,使用M‑LSD算法对目标框中的跑道轮廓线进行提取:输入跑道所在目标框图像到M‑LSD的特征提取网络中,通过特征图提取特征,对跑道轮廓进行可视化;进行线段检测,得到检测出的跑道边缘线段之后,对其进行排序拟合,按照边缘角点对跑道进行畸变校正,从而得到最终的跑道轮廓线;提取跑道轮廓线。
0/0