[发明专利]文物历史照片色彩复原方法、装置、电子设备及存储介质

权利要求书

1.一种文物历史照片色彩复原方法，其特征在于，所述方法包括：

获取待复原历史照片；

对所述待复原历史照片进行预处理，得到预处理照片；

将所述预处理照片输入预建CycleGAN网络模型，得到所述待复原历史对应的色彩复原图片。

2.根据权利要求1所述的方法，其特征在于，所述预建CycleGAN网络模型的构建步骤包括：

获取若干文物历史照片及若干现存文物彩色照片；

对所有所述文物历史照片及所述现存文物彩色照片进行预处理，对应得到预处理文物历史照片及预处理现存文物彩色照片；

根据所有所述预处理文物历史照片构建历史数据集，根据所有所述预处理现存文物彩色照片构建现存数据集；

基于所述历史数据集和所述现存数据集，训练CycleGAN网络模型，得到所述预建CycleGAN网络模型。

3.根据权利要求2所述的方法，其特征在于，所述CycleGAN网络模型采用基于注意力机制的CycleGAN网络模型。

4.根据权利要求3所述的方法，其特征在于，所述基于注意力机制的CycleGAN网络模型包括：基于注意力机制的特征图上采样网络结构和基于注意力机制的特征图下采样网络结构。

5.根据权利要求2-4任一项所述的方法，其特征在于，所述方法还包括：

根据所述预处理文物历史照片构建历史测试集；

基于所述历史测试集，对所述预建CycleGAN网络模型进行测试，评估色彩复原效果。

6.根据权利要求2-4任一项所述的方法，其特征在于，所述历史数据集和所述现存数据集中照片个数及内容均无需一一对应。

7.根据权利要求1或2所述的方法，其特征在于，所述预处理包括裁剪、缩放。

8.一种文物历史照片色彩复原装置，其特征在于，所述装置包括：

获取模块，用于获取待复原历史照片；

预处理模块，用于对所述待复原历史照片进行预处理，得到预处理照片；

处理模块，用于将所述预处理照片输入预建CycleGAN网络模型，得到所述待复原历史对应的色彩复原图片。

9.一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1-7中任一所述的文物历史照片色彩复原方法。

10.一种可读存储介质，其上存储有计算机程序，其特征在于，该程序被处理器执行时实现如权利要求1-7中任一所述的文物历史照片色彩复原方法。