ControlNet / README.md
Ezi's picture
restructuring the MC
cb79a56
|
raw
history blame
10.6 kB
metadata
license: openrail
datasets:
  - HighCWu/fill50k
pipeline_tag: text-to-image

Model Card for ControlNet

ControlNet is a neural network structure to control diffusion models by adding extra conditions.

This is the pretrained weights and some other detector weights of ControlNet.

Model Details

Model Description

The associated paper details:

ControlNet, an end-to-end neural network architecture that controls large image diffusion models (like Stable Diffusion) to learn task-specific input conditions.

  • Developed by: Lvmin Zhang and Maneesh Agrawala
  • Shared by: Lvmin Zhang
  • Model type: Text to Image
  • Language(s) (NLP): [More Information Needed]
  • License: OpeRAIL
  • Finetuned from model [optional]: [More Information Needed]

Model Sources]

Uses

Direct Use

Control pretrained large diffusion models to support additional input conditions.

Downstream Use

See Description of Files section for different models to control Stable Diffusion.

Out-of-Scope Use

The model should not be used to intentionally create or disseminate images that create hostile or alienating environments for people. This includes generating images that people would foreseeably find disturbing, distressing, or offensive; or content that propagates historical or current stereotypes.

Bias, Risks, and Limitations

[More Information Needed]

Recommendations

Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.

How to Get Started with the Model

Use the code below to get started with the model.

[More Information Needed]

Training Details

Training Data

See fill50k Dataset Card and the ControlNet/training/fill50k.zip file for the data for our training tutorial.

Training Procedure

Preprocessing [optional]

[More Information Needed]

Training Hyperparameters

  • Training regime: [More Information Needed]

Speeds, Sizes, Times [optional]

[More Information Needed]

Evaluation

Testing Data, Factors & Metrics

Testing Data

[More Information Needed]

Factors

[More Information Needed]

Metrics

[More Information Needed]

Results

[More Information Needed]

Summary

[More Information Needed]

Model Examination [optional]

[More Information Needed]

Environmental Impact

Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).

  • Hardware Type: Nvidia A100 80G
  • Hours used: 600
  • Cloud Provider: [More Information Needed]
  • Compute Region: [More Information Needed]
  • Carbon Emitted: [More Information Needed]

The associated paper detials the compute hardware and GPU hours for:

several implementations of ControlNets with different image-based conditions to control large diffusion models in various ways. The above hardware and GPU hours corresponds to the Canny Edge detector

Technical Specifications

Model Architecture and Objective

As noted in the associated Github:

It copys the weights of neural network blocks into a "locked" copy and a "trainable" copy. The "trainable" one learns your condition. The "locked" one preserves your model. Thanks to this, training with small dataset of image pairs will not destroy the production-ready diffusion models. The "zero convolution" is 1×1 convolution with both weight and bias initialized as zeros. Before training, all zero convolutions output zeros, and ControlNet will not cause any distortion. No layer is trained from scratch. You are still fine-tuning. Your original model is safe. This allows training on small-scale or even personal devices.>

Compute Infrastructure

[More Information Needed]

Hardware

[More Information Needed]

Software

Description of Files

ControlNet/models/control_sd15_canny.pth

  • The ControlNet+SD1.5 model to control SD using canny edge detection.

ControlNet/models/control_sd15_depth.pth

  • The ControlNet+SD1.5 model to control SD using Midas depth estimation.

ControlNet/models/control_sd15_hed.pth

  • The ControlNet+SD1.5 model to control SD using HED edge detection (soft edge).

ControlNet/models/control_sd15_mlsd.pth

  • The ControlNet+SD1.5 model to control SD using M-LSD line detection (will also work with traditional Hough transform).

ControlNet/models/control_sd15_normal.pth

  • The ControlNet+SD1.5 model to control SD using normal map. Best to use the normal map generated by that Gradio app. Other normal maps may also work as long as the direction is correct (left looks red, right looks blue, up looks green, down looks purple).

ControlNet/models/control_sd15_openpose.pth

  • The ControlNet+SD1.5 model to control SD using OpenPose pose detection. Directly manipulating pose skeleton should also work.

ControlNet/models/control_sd15_scribble.pth

  • The ControlNet+SD1.5 model to control SD using human scribbles. The model is trained with boundary edges with very strong data augmentation to simulate boundary lines similar to that drawn by human.

ControlNet/models/control_sd15_seg.pth

  • The ControlNet+SD1.5 model to control SD using semantic segmentation. The protocol is ADE20k.

ControlNet/annotator/ckpts/body_pose_model.pth

  • Third-party model: Openpose’s pose detection model.

ControlNet/annotator/ckpts/hand_pose_model.pth

  • Third-party model: Openpose’s hand detection model.

ControlNet/annotator/ckpts/dpt_hybrid-midas-501f0c75.pt

  • Third-party model: Midas depth estimation model.

ControlNet/annotator/ckpts/mlsd_large_512_fp32.pth

  • Third-party model: M-LSD detection model.

ControlNet/annotator/ckpts/mlsd_tiny_512_fp32.pth

  • Third-party model: M-LSD’s another smaller detection model (we do not use this one).

ControlNet/annotator/ckpts/network-bsds500.pth

  • Third-party model: HED boundary detection.

ControlNet/annotator/ckpts/upernet_global_small.pth

  • Third-party model: Uniformer semantic segmentation.

Citation

BibTeX:

@misc{https://doi.org/10.48550/arxiv.2302.05543,
  doi = {10.48550/ARXIV.2302.05543},
  
  url = {https://arxiv.org/abs/2302.05543},
  
  author = {Zhang, Lvmin and Agrawala, Maneesh},
  
  keywords = {Computer Vision and Pattern Recognition (cs.CV), Artificial Intelligence (cs.AI), Graphics (cs.GR), Human-Computer Interaction (cs.HC), Multimedia (cs.MM), FOS: Computer and information sciences, FOS: Computer and information sciences},
  
  title = {Adding Conditional Control to Text-to-Image Diffusion Models},
  
  publisher = {arXiv},
  
  year = {2023},
  
  copyright = {arXiv.org perpetual, non-exclusive license}
}

Glossary [optional]

[More Information Needed]

More Information

Special Thank to the great project - Mikubill' A1111 Webui Plugin !

We also thank Hysts for making Gradio demo in Hugging Face Space as well as more than 65 models in that amazing Colab list!

Thank haofanwang for making ControlNet-for-Diffusers!

We also thank all authors for making Controlnet DEMOs, including but not limited to fffiloni, other-model, ThereforeGames, RamAnanth1, etc!

Model Card Authors

Lvmin Zhang in collaboration with Ezi Ozoani and the Hugging Face team.

Model Card Contact

[More Information Needed]