基于模型剪枝和知识蒸馏的船舶目标检测方法
Journal of Nanjing Institute of Technology(Natural Science Edition)（2022）
摘要
针对基于深度学习的目标检测技术需要大量计算、卷积网络里大量参数占用内存空间导致模型太大无法较好进行部署等问题,提出一种基于模型剪枝和知识蒸馏的船舶目标检测方法.首先构建小样本船舶数据集,通过数据增广来扩充数据集数量,然后采用YOLO v3模型对其进行训练,接着对网络进行模型剪枝,最后采用知识蒸馏技术对剪枝后的模型进行蒸馏,补偿剪枝后模型的检测精度损失.试验表明,剪枝后压缩了模型体积,且仍有较高的检测精度.
更多AI 理解论文
溯源树
样例
生成溯源树，研究论文发展脉络
Chat Paper
正在生成论文摘要