基于自编码器与YOLO算法的高清图像小目标检测方法

作者：吴宪云; 孙力; 李云松; 王柯俨; 刘凯; 雷杰; 郭杰; 苏丽雪; 王康; 司鹏辉

来源：2020-03-04, 中国, ZL202010143805.7.

本发明公开了一种基于自编码器与YOLO算法的高清图像小目标检测方法,主要解决现有技术对高清图像小目标检测其准确率和速度不可兼顾的问题。实现步骤为：1)采集高清图像并标注,获得训练集和测试集；2)对标注好的训练集进行数据扩充；3)根据标注信息生成对应的Mask数据；4)搭建自编码器模型；5)使用训练集对其训练；6)将训练好的自编码器的编码网络与YOLO-V3检测网络拼接,得到混合网络并使用训练集对其训练；7)使用训练好的混合网络在测试集上进行目标检测。本发明减少了目标检测的计算量,提高了检测速度,并在保证检测速度的情况下提高了高清图像中小目标的检测精度,可用于无人机航拍图像的目标识别。

• 单位
  西安电子科技大学