LiamKhoaLe commited on
Commit
264c76c
·
1 Parent(s): 0a28218

enlarge dim

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.3 else "sand"
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):