一种基于深度学习的无人机关键部件实时视频语义分割方法

摘要

本发明公开了一种基于深度学习的无人机关键部件实时视频语义分割方法，属于图像的语义分割技术领域。本发明通过收集网络上所有出现无人机的场景，并对连续多帧进行拆分，标注关键帧后进行数据增强，得到无人机视频语义分割数据集，使用多分支特征提取网络，对视频帧进行特征学习，将多分支特征融合补充，达到深层网络特征提取效果，以达到相同精度下提升网络速度的效果；使用光流信息将分割阶段的低级特征对称融合到后续卷积阶段，保证边缘细节不在卷积网络中丢失，保证了边缘分割到精度，为后续利用边缘信息的中心点计算提供保证。本发明根据视频语义分割识别结果，基于像素替换和二值化处理等获取目标部件轮廓及中心点，提升了易用性。