mattmdjaga commited on
Commit
48531bf
1 Parent(s): b5ef6a6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -0
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