The image collected by the infrared night vision device is a grayscale image, but the human eye can distinguish only dozens of grayscale levels, which restricts the analysis and processing of the measurement image. Infrared image colorization can greatly improve the target tracking and recognition performance, speed and accuracy. This paper proposes an infrared coloring model based on DCGAN and the corresponding model compression algorithm. The generator of the DGCAN model adopts the Unet structure, and the discriminator adopts the PatchGAN structure to ensure better performance in the domain transfer from infrared images to color images. Hinge Loss and L1 loss are used to ensure the stability of model training, while taking into account the edge device transplantation of the model, the network is distilled during the training process, network pruning and neural network structure search and other operations to achieve model compression and deploy on edge devices superior. The experimental results of multiple datasets show that the colorization algorithm proposed in this paper can greatly improve the processing speed while maintaining the colorization performance, and realize the real-time processing of infrared image colorization.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Infrared Image Colorization Algorithm Based on DCGAN and Its Edge Device Acceleration


    Contributors:
    Chi, Haoyuan (author) / Zhang, Zebin (author) / Ge, Qiqi (author) / Zhu, Wenhuan (author)


    Publication date :

    2022-10-12


    Size :

    1830131 byte




    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English



    A Fast Colorization Algorithm for Infrared Video

    He, Mengchi / Gu, Xiaojing / Gu, Xingsheng | Springer Verlag | 2014


    A Fast Colorization Algorithm for Infrared Video

    He, M. / Gu, X. | British Library Conference Proceedings | 2014


    DCGAN Method and apparatus for recognizing vehicle using DCGAN in parking lot

    LEE JIN HEE / JEONG BO SUN | European Patent Office | 2024

    Free access

    Image colorization based on texture map

    Liu, S. / Zhang, X. | British Library Online Contents | 2013


    Colorization algorithm using probabilistic relaxation

    Horiuchi, T. | British Library Online Contents | 2004