first frame mask variable name fixed
Browse files
app.py
CHANGED
@@ -163,8 +163,8 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
|
|
163 |
state = siamese_init(
|
164 |
frame, target_pos, target_sz, siammask, cfg['hp'], device=device)
|
165 |
original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
|
166 |
-
frame[:, :, 2] = (
|
167 |
-
(
|
168 |
else:
|
169 |
# track
|
170 |
state = siamese_track(
|
|
|
163 |
state = siamese_init(
|
164 |
frame, target_pos, target_sz, siammask, cfg['hp'], device=device)
|
165 |
original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
|
166 |
+
frame[:, :, 2] = (masked_frame == 255) + \
|
167 |
+
(masked_frame == 0) * frame[:, :, 2]
|
168 |
else:
|
169 |
# track
|
170 |
state = siamese_track(
|