Commit
·
b0d277a
1
Parent(s):
70b6774
Update logic syntax
Browse files
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,
|
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.
|