[发明专利]一种基于通道剪枝的无人机图像实时目标检测方法

摘要

本发明公开了一种基于通道剪枝的无人机图像实时目标检测方法，使用预训练完的网络作为初始网络，使用更新的损失函数进行稀疏化训练，将所有批次标准化层的尺度缩放因子按顺序排列，依据稀疏化阈值对通道进行剪枝；通道剪枝过程利用掩码来标记卷积层通道，需要进行剪枝的通道掩码为1，保留的通道掩码为0；逐层地对网络层进行剪枝，根据掩码判断是否删除该通道相连的输入、输出、卷积核以及批次标准化层的参数，将待剪枝通道操作完成后生成新的模型参数文件；最后使用较小的学习率来微调通道剪枝后的模型，恢复模型的目标识别精度。本发明对硬件资源的要求低、识别速度快，能够实时识别无人机所处场景。