基于稀疏量化的图像超分辨率模型压缩方法及装置

本发明公开了一种基于稀疏量化的图像超分辨率模型压缩方法及装置，该方法包括：基于样本图像对图像超分辨率模型进行多轮迭代训练，直至满足收敛条件为止得到训练并压缩好的图像超分辨率模型；其中，图像超分辨率模型的每轮迭代训练过程均包括：选择图像超分辨率模型的部分模型权重进行二值化，以完成对模型权重的稀疏二值化处理，得到部分压缩后的模型；将样本图像输入至部分压缩后的模型，得到模型损失；根据模型损失，采用不同的梯度更新策略更新部分模型权重和未被选择的模型权重，以对模型权重进行稀疏二值化更新，得到更新后的图像超分辨率模型。经本发明提供的方法压缩后的模型存储成本和计算复杂度较低。