LiamKhoaLe commited on
Commit
b0d277a
·
1 Parent(s): 70b6774

Update logic syntax

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -117,7 +117,7 @@ def build_masks(seg):
117
  return water_mask, garbage_mask, movable_mask
118
 
119
  # Garbage mask can be highlighted in red
120
- def highlight_chunk_masks_on_frame(frame, labels, objs, num_cc, color_uncollected=(0, 0, 128), color_collected=(0, 128, 0), alpha=0.3):
121
  """
122
  Overlays semi-transparent colored regions for garbage chunks on the frame.
123
  `objs` must have 'pos' and 'col' keys. The collection status changes the overlay color.
 
117
  return water_mask, garbage_mask, movable_mask
118
 
119
  # Garbage mask can be highlighted in red
120
+ def highlight_chunk_masks_on_frame(frame, labels, objs, color_uncollected=(0, 0, 128), color_collected=(0, 128, 0), alpha=0.3):
121
  """
122
  Overlays semi-transparent colored regions for garbage chunks on the frame.
123
  `objs` must have 'pos' and 'col' keys. The collection status changes the overlay color.