[发明专利]轻量化目标图像识别方法、装置及存储介质

摘要

本发明公开了轻量化目标图像识别方法、装置及存储介质，利用训练获得的轻量化目标图像识别网络模型识别目标图像类别、位置和/或置信度；其中模型的训练方法包括：获取目标图像样本；采用预训练获取的YOLOv6‑n神经网络模型对目标图像样本进行迁移学习；对神经网络模型进行通道剪枝，获得轻量化神经网络模型；利用自适应知识蒸馏方法获得有效知识集合，利用所述有效知识集合训练所述轻量化神经网络模型，最终获得满足要求的轻量化目标图像识别网络模型。通过通道剪枝对模型进行压缩，该方法能在模型通道剪枝训练过程中识别不重要的卷积通道并进行修剪，从而获得更为紧凑的模型，并利用自适应知识蒸馏方法提高了模型的精度。