基于自编码器与YOLO算法的高清图像小目标检测方法
基本信息
| 申请号 | CN202010143805.7 | 申请日 | - |
| 公开（公告）号 | CN111126359A | 公开（公告）日 | 2020-05-08 |
| 申请公布号 | CN111126359A | 申请公布日 | 2020-05-08 |
| 分类号 | G06K9/00;G06K9/62;G06N3/04;G06N3/08 | 分类 | 计算；推算；计数; |
| 发明人 | 吴宪云;孙力;李云松;王柯俨;刘凯;雷杰;郭杰;苏丽雪;王康;司鹏辉 | 申请（专利权）人 | 南京亿芯亿意信息科技有限公司 |
| 代理机构 | 陕西电子工业专利中心 | 代理人 | 西安电子科技大学 |
| 地址 | 710071 陕西省西安市太白南路2号 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明公开了一种基于自编码器与YOLO算法的高清图像小目标检测方法，主要解决现有技术对高清图像小目标检测其准确率和速度不可兼顾的问题。实现步骤为：1)采集高清图像并标注，获得训练集和测试集；2)对标注好的训练集进行数据扩充；3)根据标注信息生成对应的Mask数据；4)搭建自编码器模型；5)使用训练集对其训练；6)将训练好的自编码器的编码网络与YOLO‑V3检测网络拼接，得到混合网络并使用训练集对其训练；7)使用训练好的混合网络在测试集上进行目标检测。本发明减少了目标检测的计算量，提高了检测速度，并在保证检测速度的情况下提高了高清图像中小目标的检测精度，可用于无人机航拍图像的目标识别。 |
