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

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