metadata

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
SAM-COYO-3k
COCO2017

Training Method

Train on ControlNet-ade20k for 30 epochs.
Train on SAM-COYO-3k for 30 epochs.
Train on COCO2017 for 1 epoch.