|
--- |
|
license: mit |
|
datasets: |
|
- Levi980623/ControlNet_ade20k |
|
- mfidabel/sam-coyo-3k |
|
- lmms-lab/COCO-Caption2017 |
|
pipeline_tag: image-to-image |
|
language: |
|
- en |
|
tags: |
|
- stable-diffusion |
|
- controlnet |
|
--- |
|
# Model Card for lavinal712/controlnet-v1-seg |
|
|
|
## Model Description |
|
|
|
This is controlnet weight trained on runwayml/stable-diffusion-v1-5 with segmentaion. |
|
|
|
### Training |
|
|
|
This model was trained using a Segmented dataset based on the ADE20k Dataset, COYO-700M Dataset and COCO Dataset. Stable Diffusion v1.5 checkpoint was used as the base model for the controlnet. |
|
|
|
- [ControlNet-ade20k](https://huggingface.co/datasets/Levi980623/ControlNet_ade20k) |
|
- [SAM-COYO-3k](https://huggingface.co/datasets/mfidabel/sam-coyo-3k) |
|
- [COCO2017](https://huggingface.co/datasets/lmms-lab/COCO-Caption2017) |
|
|
|
#### Training Method |
|
|
|
- Train on [ControlNet-ade20k](https://huggingface.co/datasets/Levi980623/ControlNet_ade20k) for 30 epochs. |
|
- Train on [SAM-COYO-3k](https://huggingface.co/datasets/mfidabel/sam-coyo-3k) for 30 epochs. |
|
- Train on [COCO2017](https://huggingface.co/datasets/lmms-lab/COCO-Caption2017) for 1 epoch. |