Update README.md
Browse files
README.md
CHANGED
|
@@ -3,6 +3,7 @@ license: mit
|
|
| 3 |
datasets:
|
| 4 |
- Levi980623/ControlNet_ade20k
|
| 5 |
- mfidabel/sam-coyo-3k
|
|
|
|
| 6 |
pipeline_tag: image-to-image
|
| 7 |
language:
|
| 8 |
- en
|
|
@@ -22,8 +23,10 @@ This model was trained using a Segmented dataset based on the ADE20k Dataset and
|
|
| 22 |
|
| 23 |
- [ControlNet-ade20k](https://huggingface.co/datasets/Levi980623/ControlNet_ade20k)
|
| 24 |
- [SAM-COYO-3k](https://huggingface.co/datasets/mfidabel/sam-coyo-3k)
|
|
|
|
| 25 |
|
| 26 |
#### Training Method
|
| 27 |
|
| 28 |
- Train on [ControlNet-ade20k](https://huggingface.co/datasets/Levi980623/ControlNet_ade20k) for 30 epochs.
|
| 29 |
- Train on [SAM-COYO-3k](https://huggingface.co/datasets/mfidabel/sam-coyo-3k) for 30 epochs.
|
|
|
|
|
|
| 3 |
datasets:
|
| 4 |
- Levi980623/ControlNet_ade20k
|
| 5 |
- mfidabel/sam-coyo-3k
|
| 6 |
+
- lmms-lab/COCO-Caption2017
|
| 7 |
pipeline_tag: image-to-image
|
| 8 |
language:
|
| 9 |
- en
|
|
|
|
| 23 |
|
| 24 |
- [ControlNet-ade20k](https://huggingface.co/datasets/Levi980623/ControlNet_ade20k)
|
| 25 |
- [SAM-COYO-3k](https://huggingface.co/datasets/mfidabel/sam-coyo-3k)
|
| 26 |
+
- [COCO2017](https://huggingface.co/datasets/lmms-lab/COCO-Caption2017)
|
| 27 |
|
| 28 |
#### Training Method
|
| 29 |
|
| 30 |
- Train on [ControlNet-ade20k](https://huggingface.co/datasets/Levi980623/ControlNet_ade20k) for 30 epochs.
|
| 31 |
- Train on [SAM-COYO-3k](https://huggingface.co/datasets/mfidabel/sam-coyo-3k) for 30 epochs.
|
| 32 |
+
- Train on [COCO2017](https://huggingface.co/datasets/lmms-lab/COCO-Caption2017) for 1 epoch.
|