File size: 1,477 Bytes
9f2fd3b 01d9795 9f2fd3b 01d9795 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
---
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] |