metadata

license: apache-2.0
datasets:
  - detection-datasets/coco
language:
  - en
library_name: diffusers
tags:
  - pytorch
  - controlnet
  - image-colorization
  - image-to-image
pipeline_tag: image-to-image

Model Card for ColorizeNet

This model is a ControlNet training to perform image colorization from black and white images.

Model Details

Model Description

ColorizeNet is an image colorization model based on ControlNet, trained using the pre-trained Stable Diffusion model version 2.1 proposed by Stability AI.

Finetuned from model : [https://huggingface.co/stabilityai/stable-diffusion-2-1]

Model Sources [optional]

Repository: [https://github.com/rensortino/ColorizeNet]

How to Get Started with the Model

Use the code below to get started with the model.

[More Information Needed]

Training Details

Training Data

The model has been trained on COCO, using all the images in the dataset and converting them to grayscale to use them to condition the ControlNet

[https://huggingface.co/datasets/detection-datasets/coco]

Results

[More Information Needed]