-
公开(公告)号:CN116433508A
公开(公告)日:2023-07-14
申请号:CN202310255192.X
申请日:2023-03-16
Applicant: 湖北大学
Abstract: 本发明公开一种基于Swin‑Unet的灰度图像着色校正方法,包括:获取原始图像,对所述原始图像预处理得到预处理数据集,其中,所述预处理数据集包括训练集和测试集;构建灰度图像着色校正模型,将所述训练集输入到所述灰度图像着色校正模型进行训练,得到训练好的灰度图像着色校正模型;将所述测试集输入到所述训练好的灰度图像着色校正模型,得到最终的彩色图片。本发明利用Swin‑Unet对灰度图像进行颜色预测,并通过卷积神经网络对预估颜色进行颜色校正,确保颜色校正效果,并且泛化性强。
-
公开(公告)号:CN118172431B
公开(公告)日:2024-11-05
申请号:CN202410235479.0
申请日:2024-03-01
Applicant: 湖北大学
IPC: G06T11/00 , G06T7/90 , G06N3/0455 , G06V10/44 , G06V10/82
Abstract: 本发明公开了一种基于先验索引的颜色标记重加权的灰度图像着色方法,包括:获取连续灰度图像特征,获取连续彩色图像特征并编码获得离散彩色图像特征;获取双向自注意力transformer网络,结合连续灰度图像特征、离散彩色图像特征获得颜色标记;统计真实数据集中不同亮度范围的颜色概率分布,结合灰度图像的亮度分布信息获得颜色候选值,基于颜色候选值获得合成图像,基于合成图像获得候选逻辑值的索引图,根据对应的灰度掩码选取候选的逻辑值索引,采用加权算法控制颜色标记的权重,获得最优颜色标记;将最优颜色标记和连续灰度图像特征解码实现灰度图像着色。
-
公开(公告)号:CN118172431A
公开(公告)日:2024-06-11
申请号:CN202410235479.0
申请日:2024-03-01
Applicant: 湖北大学
IPC: G06T11/00 , G06T7/90 , G06N3/0455 , G06V10/44 , G06V10/82
Abstract: 本发明公开了一种基于先验索引的颜色标记重加权的灰度图像着色方法,包括:获取连续灰度图像特征,获取连续彩色图像特征并编码获得离散彩色图像特征;获取双向自注意力transformer网络,结合连续灰度图像特征、离散彩色图像特征获得颜色标记;统计真实数据集中不同亮度范围的颜色概率分布,结合灰度图像的亮度分布信息获得颜色候选值,基于颜色候选值获得合成图像,基于合成图像获得候选逻辑值的索引图,根据对应的灰度掩码选取候选的逻辑值索引,采用加权算法控制颜色标记的权重,获得最优颜色标记;将最优颜色标记和连续灰度图像特征解码实现灰度图像着色。
-
公开(公告)号:CN116433508B
公开(公告)日:2023-10-27
申请号:CN202310255192.X
申请日:2023-03-16
Applicant: 湖北大学
Abstract: 本发明公开一种基于Swin‑Unet的灰度图像着色校正方法,包括:获取原始图像,对所述原始图像预处理得到预处理数据集,其中,所述预处理数据集包括训练集和测试集;构建灰度图像着色校正模型,将所述训练集输入到所述灰度图像着色校正模型进行训练,得到训练好的灰度图像着色校正模型;将所述测试集输入到所述训练好的灰度图像着色校正模型,得到最终的彩色图片。本发明利用Swin‑Unet对灰度图像进行颜色预测,并通过卷积神经网络对预估颜色进行颜色校正,确保颜色校正效果,并且泛化性强。
-
-
-