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