图像重构模型的训练方法、图像重构方法及系统

摘要

本发明提供一种图像重构模型的训练方法、图像重构方法及系统，训练方法包括：获取第一训练集；对于第一训练集中每组训练样本对，将有油烟样本图像输入特征提取子模型进行特征提取，得到初始特征图；将初始特征图输入空间注意力子模型得到包含油烟区域信息的注意力图；将初始特征图和注意力图相乘得到包含油烟区域信息的空间特征图；将空间特征图输入生成器得到重构图像；将重构图像输入判别器以判断重构图像是否为无油烟图像；根据无油烟样本图像、判别器的输出结果以及重构图像计算损失误差，并根据损失误差调整模型参数，直至图像重构模型符合迭代停止条件。本发明实现了油烟区域的特征相互联系，重构图像能够符合真实烹饪场景。