无人机平台多尺度目标检测方法和装置

基本信息

申请号 CN202110225622.4 申请日 -
公开(公告)号 CN112819100A 公开(公告)日 2021-05-18
申请公布号 CN112819100A 申请公布日 2021-05-18
分类号 G06K9/62;G06K9/46;G06K9/00 分类 计算;推算;计数;
发明人 王斌;金帅;王礼健 申请(专利权)人 深圳中湾智能科技有限公司
代理机构 北京众元弘策知识产权代理事务所(普通合伙) 代理人 李超
地址 518055 广东省深圳市南山区桃源街道平山社区留仙大道1213号红花岭工业区第五区B栋502
法律状态 -

摘要

摘要 本发明公开了一种无人机平台多尺度目标检测方法,使用无监督学习对训练集的目标框进行聚类,引入空洞卷积,通过网络对多尺度目标特征进行提取,加入正负样本平衡的方式;在训练阶段,加入图像增强技术;使用正则化方法,进一步优化网络结构;在后处理阶段,使用soft‑NMS替换NMS,解决目标检测中的物体重叠难以预测,分类分数与IoU不匹配的问题;引入注意力机制,对通道间的依赖关系进行建模,自适应调整各通道的特征响应值。本发明基于YOLOV3算法进行改进,可以满足本案的多尺度、实时性高、精度高的要求,在模型训练阶段引入图像增强技术和正则化方法提升算法和网络结构的模型泛化性能和鲁棒性。