Image colorization provides significant data handling advantages by completely eliminating the need for labeling. This work introduces advancements in the process of converting grayscale images into color using modern machine learning techniques. By utilizing the Lab color space, where luminance (L) is processed separately from the color channels (ab), this article focuses on refining and developing the Conditional Variational Autoencoder (CVAE) model to address the challenges in image colorization with the LFW dataset. Additionally, the Weights & Biases (wandb) tool is employed to support the monitoring of the training process. The training results indicate that the CVAE model achieves commendable outcomes, with a Peak Signal-to-Noise Ratio (PSNR) of 27.63 dB and a Structural Similarity Index Measure (SSIM) of 91.97%. These results highlight the potential of the CVAE model for effective and efficient image colorization.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Optimizing Image Colorization: Innovations with CVAE

  • Ngoc-Giau Pham,
  • Van-Hieu Duong,
  • Thanh-Hai Tong-Le,
  • Khac-Hoang Nguyen,
  • Phu-Thanh Ha-Quach

摘要

Image colorization provides significant data handling advantages by completely eliminating the need for labeling. This work introduces advancements in the process of converting grayscale images into color using modern machine learning techniques. By utilizing the Lab color space, where luminance (L) is processed separately from the color channels (ab), this article focuses on refining and developing the Conditional Variational Autoencoder (CVAE) model to address the challenges in image colorization with the LFW dataset. Additionally, the Weights & Biases (wandb) tool is employed to support the monitoring of the training process. The training results indicate that the CVAE model achieves commendable outcomes, with a Peak Signal-to-Noise Ratio (PSNR) of 27.63 dB and a Structural Similarity Index Measure (SSIM) of 91.97%. These results highlight the potential of the CVAE model for effective and efficient image colorization.