Downloads · 30 days
0
polygon-colorization-unet
aashu-0/polygon-colorization-unet
polygon-colorization-unet is a image-to-image model from aashu-0. Use it when you need one image transformed into another. The card lists the license as mit.
Conditional UNet model trained to colorize polygon images based on color names.
Access
Public
Updated Aug 5, 2025
Repo size
208 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth208 MB · 100%
At a glance
- Task
- Image-to-Image
- License
- mit
- Access
- Public
- Created
- Aug 5, 2025
- Updated
- Aug 5, 2025
- SHA
- ab757f7f
Model card
From the Hugging Face model README
Conditional UNet model trained to colorize polygon images based on color names.
Model Details
- Task: Image-to-Image (Polygon Colorization)
- Architecture: Conditional UNet
- Framework: PyTorch
- Input: Polygon image (256x256 RGB) + Color name
- Output: Colored polygon image (256x256 RGB)
Supported Colors
red, green, blue, yellow, orange, purple, cyan, magenta
- Task
- Image-to-Image
- License
- mit
- Created
- Aug 5, 2025
- Updated
- Aug 5, 2025