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.