ELECTRONIC DEVICE FOR COLORIZING BLACK AND WHITE IMAGE USING GAN BASED MODEL COMPRISING TRANSFORMER BLOCK AND METHOD FOR OPERATION THEREOF

    公开(公告)号:US20240265587A1

    公开(公告)日:2024-08-08

    申请号:US18422534

    申请日:2024-01-25

    IPC分类号: G06T11/00

    CPC分类号: G06T11/001

    摘要: In accordance with various embodiments, an electronic device for colorizing a black and white image using a Generative Adversarial Network (GAN)-based model comprising a transformer block includes a processor, wherein the processor is set to: obtain a black and white image including only first information about a luminance channel; and generate a pseudo color image including only second information about a chrominance channel by applying the black and white image to the GAN-based model, the GAN-based model includes a generator network including a plurality of transformer blocks for color conversion, a plurality of convolution layers, and a plurality of transpose convolution layers, the plurality of transformer blocks each include a Depth Wise Convolution (DWC) layer, a first Layer Normalization (LN) layer, a Window-based Multi-head Self Attention (W-MSA) layer, a second LN layer, and a Colorization Feed Forward (CFF) block. Other various embodiments are possible.