Hugging Face's logo

mattmdjaga
/
segformer_b2_clothes

Image Segmentation
Transformers
PyTorch
ONNX
Safetensors
segformer
vision
Inference Endpoints
Model card Files Community
20
mattmdjaga commited on
Commit
de509b2
•
1 Parent(s): d7274fc

Added labels to README

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -44,6 +44,10 @@ upsampled_logits = nn.functional.interpolate(
44
  pred_seg = upsampled_logits.argmax(dim=1)[0]
45
  plt.imshow(pred_seg)
46
  ```
 
 
 
 
47
  ### License
48
 
49
  The license for this model can be found [here](https://github.com/NVlabs/SegFormer/blob/master/LICENSE).
 
44
  pred_seg = upsampled_logits.argmax(dim=1)[0]
45
  plt.imshow(pred_seg)
46
  ```
47
+
48
+ Labels: "0": "Background", "1": "Hat", "2": "Hair", "3": "Sunglasses", "4": "Upper-clothes", "5": "Skirt", "6": "Pants", "7": "Dress", "8": "Belt", "9": "Left-shoe", "10": "Right-shoe", "11": "Face", "12": "Left-leg", "13": "Right-leg", "14": "Left-arm", "15": "Right-arm", "16": "Bag", "17": "Scarf"
49
+
50
+
51
  ### License
52
 
53
  The license for this model can be found [here](https://github.com/NVlabs/SegFormer/blob/master/LICENSE).