Commit
•
48531bf
1
Parent(s):
b5ef6a6
Update README.md
Browse files
README.md
CHANGED
|
@@ -16,6 +16,11 @@ datasets:
|
|
| 16 |
SegFormer model fine-tuned on [ATR dataset](https://github.com/lemondan/HumanParsing-Dataset) for clothes segmentation but can also be used for human segmentation.
|
| 17 |
The dataset on hugging face is called "mattmdjaga/human_parsing_dataset".
|
| 18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
```python
|
| 20 |
from transformers import SegformerImageProcessor, AutoModelForSemanticSegmentation
|
| 21 |
from PIL import Image
|
|
|
|
| 16 |
SegFormer model fine-tuned on [ATR dataset](https://github.com/lemondan/HumanParsing-Dataset) for clothes segmentation but can also be used for human segmentation.
|
| 17 |
The dataset on hugging face is called "mattmdjaga/human_parsing_dataset".
|
| 18 |
|
| 19 |
+
|
| 20 |
+
**NEW** -
|
| 21 |
+
**[Training code](https://github.com/mattmdjaga/segformer_b2_clothes)**. Right now it only contains the pure code with some comments, but soon I'll add a colab notebook version
|
| 22 |
+
and a blog post with it to make it more friendly.
|
| 23 |
+
|
| 24 |
```python
|
| 25 |
from transformers import SegformerImageProcessor, AutoModelForSemanticSegmentation
|
| 26 |
from PIL import Image
|