一种基于SwinTransformer的去马赛克重建彩色图像的方法

    公开(公告)号:CN115601240A

    公开(公告)日:2023-01-13

    申请号:CN202211347627.5

    申请日:2022-10-31

    申请人: 吉林大学(CN)

    IPC分类号: G06T3/40 G06N3/04 G06N3/08

    摘要: 本发明适用于图像处理和深度学习领域,提供了一种基于SwinTransformer的去马赛克重建彩色图像的方法,包括以下步骤:1)构建数据集;2)采用基于SwinTransformer的图像恢复网络SwinIR,构建去马赛克模型;3)训练去马赛克模型;4)测试得到的去马赛克网络模型的性能。本发明提供了一种基于Swin Transformer的自注意力机制能够从Bayer图像重建RGB图像,对比卷积神经网络(CNN)的静态权重以及长程依赖,好处在于能够扩大感受野,注意到全局上下文的特征,对于纹理边缘细节也有更好的重建效果。