一种基于通道裁剪的无人机视频小目标检测方法

基本信息

申请号 CN202111567636.0 申请日 -
公开(公告)号 CN114220032A 公开(公告)日 2022-03-22
申请公布号 CN114220032A 申请公布日 2022-03-22
分类号 G06V20/17(2022.01)I;G06V20/40(2022.01)I;G06V10/82(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 贾海涛;邹新雷;周焕来;王云;乔磊崖;陈璐;孙靖哲;王俊 申请(专利权)人 一拓通信集团股份有限公司
代理机构 - 代理人 -
地址 210000江苏省南京市江苏省自贸区南京片区浦滨路320号科创一号大厦B座701室
法律状态 -

摘要

摘要 本发明公开了一种基于通道裁剪的无人机视频小目标检测方法。该发明在小目标检测方向上具有一定的通用性,该专利以千米无人机高空检测为说明案例。市区以及郊区中存在大量的小目标。针对小目标问题,在通常目标检测方法中为了减小模型内存占用、节省计算资源,根据目标尺寸特点,对YOLOv4原模型的预测层进行了改进,将三尺度检测模型改进为双尺度检测模型;对双尺度检测模型进行正常训练,然后将其BN层的缩放因子进行稀疏训练,最后通过裁剪一定比例的通道数以再次减小模型内存占用提升检测速度。基于通道裁剪的无人机视频小目标检测方法模型能够快速精准的检测远距离小目标,且对于小目标缺陷和极端长宽比缺陷也能够取得较好的效果。