pirocheto commited on
Commit
1ddc548
·
1 Parent(s): a54fa65

fix: README label table layout

Browse files
Files changed (1) hide show
  1. README.md +9 -13
README.md CHANGED
@@ -42,18 +42,14 @@ ATR is a large single-person human parsing dataset with 17 000+ images focused o
42
 
43
  ## Labels
44
 
45
- | ID | Label | ID | Label |
46
- |----|-------|----|-------|
47
- | 0 | Background | 10 | Right-shoe |
48
- | 1 | Hat | 11 | Face |
49
- | 2 | Hair | 12 | Left-leg |
50
- | 3 | Sunglasses | 13 | Right-leg |
51
- | 4 | Upper-clothes | 14 | Left-arm |
52
- | 5 | Skirt | 15 | Right-arm |
53
- | 6 | Pants | 16 | Bag |
54
- | 7 | Dress | 17 | Scarf |
55
- | 8 | Belt | | |
56
- | 9 | Left-shoe | | |
57
 
58
  ## Usage — PyTorch
59
 
@@ -62,7 +58,7 @@ from transformers import AutoImageProcessor, AutoModelForSemanticSegmentation
62
  from PIL import Image
63
  import torch
64
 
65
- model = AutoModelForSemanticSegmentation.from_pretrained("pirocheto/schp-atr-18", trust_remote_code=True)
66
  processor = AutoImageProcessor.from_pretrained("pirocheto/schp-atr-18", trust_remote_code=True)
67
 
68
  image = Image.open("photo.jpg").convert("RGB")
 
42
 
43
  ## Labels
44
 
45
+ | ID | Label | ID | Label | ID | Label |
46
+ |----|-------|----|---------|----|-------|
47
+ | 0 | Background | 6 | Pants | 12 | Left-leg |
48
+ | 1 | Hat | 7 | Dress | 13 | Right-leg |
49
+ | 2 | Hair | 8 | Belt | 14 | Left-arm |
50
+ | 3 | Sunglasses | 9 | Left-shoe | 15 | Right-arm |
51
+ | 4 | Upper-clothes | 10 | Right-shoe | 16 | Bag |
52
+ | 5 | Skirt | 11 | Face | 17 | Scarf |
 
 
 
 
53
 
54
  ## Usage — PyTorch
55
 
 
58
  from PIL import Image
59
  import torch
60
 
61
+ model = AutoModelForSemanticSegmentation.from_pretrained("pirocheto/schp-atr-18", trust_remote_code=True)
62
  processor = AutoImageProcessor.from_pretrained("pirocheto/schp-atr-18", trust_remote_code=True)
63
 
64
  image = Image.open("photo.jpg").convert("RGB")