Image colorization provides significant data handling advantages by completely eliminating the need for labeling. This research 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 research focuses on refining and developing the Dif-EDUNet model (a Diffusion model using ED-UNet) to address the challenges in image colorization. Our experiments, including training with datasets: Coco-Stuff, DIV2K, Places365, ImageNet and CelebA show that our results are very encouraging compared to the benchmark of the dataset in this issue. Additionally, the Weights & Biases (wandb) tool is employed to support the monitoring of the training process.

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

Image Colorization with Dif-EDUNet: A Diffusion-Based Approach

  • Ngoc-Giau Pham,
  • Van-Hieu Duong,
  • Thanh-Hai Le Tong,
  • Hong-Ngoc Tran,
  • Phuoc-Hung Vo

摘要

Image colorization provides significant data handling advantages by completely eliminating the need for labeling. This research 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 research focuses on refining and developing the Dif-EDUNet model (a Diffusion model using ED-UNet) to address the challenges in image colorization. Our experiments, including training with datasets: Coco-Stuff, DIV2K, Places365, ImageNet and CelebA show that our results are very encouraging compared to the benchmark of the dataset in this issue. Additionally, the Weights & Biases (wandb) tool is employed to support the monitoring of the training process.