yeonmorae commited on
Commit
a129678
1 Parent(s): 0e6910e
Files changed (3) hide show
  1. .gitattributes +1 -1
  2. app.py +19 -20
  3. labels.txt +18 -18
.gitattributes CHANGED
@@ -32,4 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
32
  *.xz filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
- *tfevents* filter=lfs diff=lfs merge=lfs -text
 
32
  *.xz filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -14,31 +14,29 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
14
  "mattmdjaga/segformer_b2_clothes"
15
  )
16
 
17
-
18
  def ade_palette():
19
  """ADE20K palette that maps each class to RGB values."""
20
  return [
21
- [255, 0, 0],
22
- [255, 145, 0],
23
- [255, 251, 0],
24
- [132, 255, 0],
25
- [0, 255, 64],
26
- [0, 255, 174],
27
- [0, 229, 255],
28
- [0, 128, 255],
29
- [0, 30, 255],
30
- [85, 0, 255],
31
- [183, 0, 255],
32
- [255, 0, 230],
33
- [255, 0, 115],
34
- [133, 48, 48],
35
- [133, 122, 48],
36
- [89, 133, 48],
37
- [48, 133, 88],
38
- [48, 76, 133],
39
  ]
40
 
41
-
42
  labels_list = []
43
 
44
  with open(r'labels.txt', 'r') as fp:
@@ -108,4 +106,5 @@ demo = gr.Interface(fn=sepia,
108
  examples=["person-1.jpg", "person-2.jpg", "person-3.jpg", "person-4.jpg", "person-5.jpg"],
109
  allow_flagging='never')
110
 
 
111
  demo.launch()
 
14
  "mattmdjaga/segformer_b2_clothes"
15
  )
16
 
 
17
  def ade_palette():
18
  """ADE20K palette that maps each class to RGB values."""
19
  return [
20
+ [204, 87, 92],
21
+ [112, 185, 212],
22
+ [45, 189, 106],
23
+ [234, 123, 67],
24
+ [78, 56, 123],
25
+ [210, 32, 89],
26
+ [90, 180, 56],
27
+ [155, 102, 200],
28
+ [33, 147, 176],
29
+ [255, 183, 76],
30
+ [67, 123, 89],
31
+ [190, 60, 45],
32
+ [134, 112, 200],
33
+ [56, 45, 189],
34
+ [200, 56, 123],
35
+ [87, 92, 204],
36
+ [120, 56, 123],
37
+ [45, 78, 123]
38
  ]
39
 
 
40
  labels_list = []
41
 
42
  with open(r'labels.txt', 'r') as fp:
 
106
  examples=["person-1.jpg", "person-2.jpg", "person-3.jpg", "person-4.jpg", "person-5.jpg"],
107
  allow_flagging='never')
108
 
109
+
110
  demo.launch()
labels.txt CHANGED
@@ -1,18 +1,18 @@
1
- background
2
- hat
3
- hair
4
- sunglasses
5
- upper-clothes
6
- skirt
7
- pants
8
- dress
9
- belt
10
- left-shoe
11
- right-shoe
12
- face
13
- left-leg
14
- right-leg
15
- left-arm
16
- right-arm
17
- bag
18
- scarf
 
1
+ Background
2
+ Hat
3
+ Hair
4
+ Sunglasses
5
+ Upper-clothes
6
+ Skirt
7
+ Pants
8
+ Dress
9
+ Belt
10
+ Left-shoe
11
+ Right-leg
12
+ Face
13
+ Left-leg
14
+ Right-leg
15
+ Left-arm
16
+ Right-arm
17
+ Bag
18
+ Scarf