基于视频的无人机识别与跟踪方法及装置

基本信息

申请号 CN202010231230.4 申请日 -
公开(公告)号 CN111460968A 公开(公告)日 2020-07-28
申请公布号 CN111460968A 申请公布日 2020-07-28
分类号 G06K9/00;G06K9/62;G06N3/04;G06N3/08 分类 -
发明人 安平;朱润杰;尤志翔;王嶺;高伟 申请(专利权)人 上海文广科技(集团)有限公司
代理机构 上海恒慧知识产权代理事务所(特殊普通合伙) 代理人 上海大学;上海文广科技(集团)有限公司
地址 200444 上海市宝山区上大路99号
法律状态 -

摘要

摘要 本发明提供了一种基于视频的无人机识别与跟踪方法及装置,该方法包括:对收集的数据集逐个进行手工无人机的标注,获得多个型号且不同尺寸的无人机标注样本;以此数据集对基于YOLOv3的网络进行训练,得到训练后的深度学习目标检测模型;采用Retinex图像增强手段提高待检测的无人机视频的图像质量,通过深度学习目标检测模型识别待检测的无人机视频每一帧;基于Sort算法实现快速对视频中的无人机跟踪。本发明可以高鲁棒高精度识别视频中的无人机并跟踪无人机,并且当无人机图像不清晰时,可以进行图像增强,适用于各种复杂场景。