基于遥感图像的YOLOv3模型算法优化

YOLOV3 FUSION PRUNING BASED ON REMOTE SENSING IMAGE

  • 摘要: 为了使YOLOv3算法能在遥感设备上实时地进行目标检测,模型压缩是常用的解决方案。根据模型压缩后精度会有损失的特点,先为模型添加注意力机制算法再对模型进行训练,并以此为基础提出基于融合卷积层与BatchNormal层后对模型进行通道剪枝方法,对YOLOv3进行通道剪枝,得到压缩后的YOLOv3目标检测模型,并对剪枝后的模型进行微调后,恢复模型的精确度。通过实验得到的结果,融合后剪枝的方法在mAP损失 0.6%的基础上,对YOLOv3的模型大小压缩94.93%,检测速度提升了150.6%;通过实验证明,该模型能够运用到对于实时性较高和检测精度较高的遥感图像目标检测,并且适用于存储空间较小的遥感设备。

     

    Abstract: In order to enable the YOLOv3 algorithm to perform real-time target detection on remote sensing equipment, model compression is a common solution. According to the characteristics of the loss of accuracy after the model is compressed, this paper added the attention mechanism algorithm to the model and trained the model. On this basis, we proposed a channel pruning method based on the fusion of the convolutional layer and the BatchNormal layer. Channel pruning was performed on YOLOv3, and we obtained the compressed YOLOv3 target detection model. After fine-tuning the pruned model, the accuracy of the model was restored. The experimental results show that the method of pruning after fusion in this paper reduces the size of the YOLOv3 model by 94.93% and increases the detection speed by 150.6% with only 0.6% mAP loss. The experiment proves that the model can be applied to real-time performance remote sensing image target detection with higher and higher detection accuracy, and suitable for remote sensing equipment with small storage space.

     

/

返回文章
返回