Commit
·
264c76c
1
Parent(s):
0a28218
enlarge dim
Browse files
app.py
CHANGED
@@ -101,7 +101,7 @@ def interpret_rgb_class(decoded_img):
|
|
101 |
ambiguous_rgb = np.array([150, 5, 61])
|
102 |
matches = np.all(np.abs(decoded_img - ambiguous_rgb) <= TOL, axis=-1)
|
103 |
match_ratio = np.count_nonzero(matches) / matches.size
|
104 |
-
return "garbage" if match_ratio > 0.
|
105 |
|
106 |
# Masking zones (Garbage and Water zone to be travelable)
|
107 |
def build_masks(seg):
|
|
|
101 |
ambiguous_rgb = np.array([150, 5, 61])
|
102 |
matches = np.all(np.abs(decoded_img - ambiguous_rgb) <= TOL, axis=-1)
|
103 |
match_ratio = np.count_nonzero(matches) / matches.size
|
104 |
+
return "garbage" if match_ratio > 0.15 else "sand"
|
105 |
|
106 |
# Masking zones (Garbage and Water zone to be travelable)
|
107 |
def build_masks(seg):
|