Spaces:

BinKhoaLe1812
/

Sall-eGarbageDetection

Running

LiamKhoaLe commited on 4 days ago

Commit

f214779

1 Parent(s): b92f45b

Update segm labelling

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,8 +84,8 @@ ade_palette = np.array([
 ], dtype=np.uint8)
 custom_class_map = {
-    "Garbage":                 [(255, 8, 41)],
-    "Water":                   [(0, 102, 200), (11, 102, 255), (31, 0, 255)],
     "Grass / Vegetation":      [(10, 255, 71), (143, 255, 140)],
     "Tree / Natural Obstacle": [(4, 200, 3), (235, 12, 255), (255, 6, 82), (255, 163, 0)],
     "Sand / Soil / Ground":    [(80, 50, 50), (230, 230, 230)],

 ], dtype=np.uint8)
 custom_class_map = {
+    "Garbage":                 [(255, 8, 41), (235, 255, 7)],
+    "Water":                   [(0, 102, 200), (11, 102, 255), (31, 0, 255), (10, 0, 255)],
     "Grass / Vegetation":      [(10, 255, 71), (143, 255, 140)],
     "Tree / Natural Obstacle": [(4, 200, 3), (235, 12, 255), (255, 6, 82), (255, 163, 0)],
     "Sand / Soil / Ground":    [(80, 50, 50), (230, 230, 230)],