AmmarAhm3d
/

colorization-model-unet

image-colorization

image-processing

computer-vision

huggingface-space

Model card Files Files and versions

AmmarAhm3d commited on Dec 16, 2025

Commit

b3825c7

·

verified ·

1 Parent(s): 73e759c

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -1,11 +1,11 @@
 ---
-license: mit
 language:
 - en
 pipeline_tag: image-to-image
 library_name: pytorch
 datasets:
-- coco
 metrics:
 - psnr
 - ssim
@@ -27,7 +27,7 @@ tags:
 ### Model Description
-`colorize-unet-pytorch` is a **PyTorch U‑Net** model for **automatic image colorization** using the **LAB color space**.
 It takes the **L\*** (grayscale/lightness) channel as input and predicts the **a\*** and **b\*** chrominance channels, which are then combined with the original L\* channel and converted back to RGB to produce a final colorized image.
 - **Developed by:** Ammar Ahmed, Ahmad Naeem, Khurram Imran

 ---
+license: unlicense
 language:
 - en
 pipeline_tag: image-to-image
 library_name: pytorch
 datasets:
+- rafaelpadilla/coco2017
 metrics:
 - psnr
 - ssim
 ### Model Description
+`colorization-model-unet` is a **PyTorch U‑Net** model for **automatic image colorization** using the **LAB color space**.
 It takes the **L\*** (grayscale/lightness) channel as input and predicts the **a\*** and **b\*** chrominance channels, which are then combined with the original L\* channel and converted back to RGB to produce a final colorized image.
 - **Developed by:** Ammar Ahmed, Ahmad Naeem, Khurram Imran