license:mit modality:image region:us
---
license: mit
---
<div align="center">

# Pre-Annotated Dataset of Shape-Guided ControlNet


[[`GitHub`]](https://github.com/AlonzoLeeeooo/shape-guided-controlnet) / [[`Model Weights`]](https://huggingface.co/AlonzoLeeeooo/shape-guided-controlnet)
</div>

Pre-annotated dataset of the re-implemented ControlNet with shape masks.

# Details
This dataset contains 82,783 shape masks, where the filename of each mask corresponds to its source image in the COCO 2014 training set.
All masks are annotated using an off-the-shelf saliency detection model, i.e., `U2-Net`.
For detailed usage of this dataset, please refer to the instructions in this [GitHub repo](https://github.com/AlonzoLeeeooo/shape-guided-controlnet).