mattmdjaga
/

segformer_b2_clothes

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

mattmdjaga commited on Mar 30, 2023

Commit

c4d1ead

·

1 Parent(s): 07f955f

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -13,7 +13,8 @@ datasets:
 ---
 # Segformer B2 fine-tuned for clothes segmentation
-SegFormer model fine-tuned on [ATR dataset](https://github.com/lemondan/HumanParsing-Dataset) for clothes segmentation
 ```python
 from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation

 ---
 # Segformer B2 fine-tuned for clothes segmentation
+SegFormer model fine-tuned on [ATR dataset](https://github.com/lemondan/HumanParsing-Dataset) for clothes segmentation.
+The dataset on hugging face is called "mattmdjaga/human_parsing_dataset"
 ```python
 from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation