[发明专利]一种基于YOLOv3的空中无人机目标识别和跟踪方法

摘要

本发明涉及视频目标识别跟踪技术领域，提供一种基于YOLOv3的空中无人机目标识别和跟踪方法。首先制作训练集；然后基于K‑means聚类算法改进候选框参数，并训练YOLOv3模型；接着获取待跟踪无人机飞行视频，设定目标模板；再生成当前帧的K*近邻搜索区域，利用YOLOv3模型检测输出类别及多个目标候选框，并计算与目标模板类别相同的目标候选框同目标模板的相似度得分，选取相似度得分最高的目标候选框作为跟踪目标，在无同类候选框时以上一帧跟踪目标为当前帧跟踪目标；最后，在未加载完图像时判断是否满足目标模板更新条件，在满足时更新目标模板。本发明能够提高空中无人机目标识别和跟踪的精度、效率和计算复杂度。