基于模型剪枝和知识蒸馏的船舶目标检测方法

作者：邱春; 路红; 秦彬鑫; 万文明; 戴云周

来源：南京工程学院学报(自然科学版), 2022, 20(02): 13-17.

DOI：10.13960/j.issn.1672-2558.2022.02.003

• 深度学习
• 卷积网络
• 模型剪枝
• 知识蒸馏

针对基于深度学习的目标检测技术需要大量计算、卷积网络里大量参数占用内存空间导致模型太大无法较好进行部署等问题，提出一种基于模型剪枝和知识蒸馏的船舶目标检测方法.首先构建小样本船舶数据集，通过数据增广来扩充数据集数量，然后采用YOLO v3模型对其进行训练，接着对网络进行模型剪枝，最后采用知识蒸馏技术对剪枝后的模型进行蒸馏，补偿剪枝后模型的检测精度损失.试验表明，剪枝后压缩了模型体积，且仍有较高的检测精度.

• 单位
  南京工程学院