一种基于模型剪枝的星载目标识别方法及装置

本发明公开了一种基于模型剪枝的星载目标识别方法及装置，包括：获取神经网络YOLOv5模型，神经网络YOLOv5模型实时识别图像或视频中的对象；对神经网络YOLOv5模型进行剪枝和重训练，剪枝包括结构化剪枝、非结构化剪枝；将经过处理的神经网络YOLOv5模型部署至星载计算机，加载遥感卫星云层图像训练数据集进行训练并推理，对推理结果进行解析，识别出卫星云图。通过本发明，能够保留YOLOv5模型的精简结构，且实现深度学习神经网络YOLOv5模型冗余参数的细粒度优化，减少神经网络YOLOv5模型参数量，从而降低计算量，保证深度学习神经网络YOLOv5模型在有限算力星载计算机中运行推理，识别出卫星云图。