Update README.md
Browse files
README.md
CHANGED
|
@@ -17,6 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 17 |
## Model description
|
| 18 |
This model classifies an image into a fashion collection. It is trained on the [tonyassi/vogue-runway-top15-512px-nobg](https://huggingface.co/datasets/tonyassi/vogue-runway-top15-512px-nobg) dataset and fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k).
|
| 19 |
|
|
|
|
|
|
|
| 20 |
## Dataset description
|
| 21 |
[tonyassi/vogue-runway-top15-512px-nobg](https://huggingface.co/datasets/tonyassi/vogue-runway-top15-512px-nobg)
|
| 22 |
- 15 fashion houses
|
|
|
|
| 17 |
## Model description
|
| 18 |
This model classifies an image into a fashion collection. It is trained on the [tonyassi/vogue-runway-top15-512px-nobg](https://huggingface.co/datasets/tonyassi/vogue-runway-top15-512px-nobg) dataset and fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k).
|
| 19 |
|
| 20 |
+
Because the model trained on a dataset with white background it is suggested to only give the model an image with a white background. Removing the background allows the model to focus on the clothes and disregard the background.
|
| 21 |
+
|
| 22 |
## Dataset description
|
| 23 |
[tonyassi/vogue-runway-top15-512px-nobg](https://huggingface.co/datasets/tonyassi/vogue-runway-top15-512px-nobg)
|
| 24 |
- 15 fashion houses
|