---
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

<!-- Provide a quick summary of what the model is/does. -->

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

## Model Details

### Model Description

<!-- Provide a longer summary of what this model is. -->

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]

<!-- Provide the basic links for the model. -->

- **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

<!-- This should link to a Data Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->

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]