color fix on siammask output
Browse files
app.py
CHANGED
@@ -175,8 +175,7 @@ def track_and_mask(vid, original_frame, masked_frame):
|
|
175 |
# track
|
176 |
state = siamese_track(
|
177 |
state, frame, mask_enable=True, refine_enable=True, device=device)
|
178 |
-
|
179 |
-
original_frame_list.append(frame)
|
180 |
location = state['ploygon'].flatten()
|
181 |
mask = state['mask'] > state['p'].seg_thr
|
182 |
frame[:, :, 2] = (mask > 0) * 255 + \
|
|
|
175 |
# track
|
176 |
state = siamese_track(
|
177 |
state, frame, mask_enable=True, refine_enable=True, device=device)
|
178 |
+
original_frame_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
|
|
|
179 |
location = state['ploygon'].flatten()
|
180 |
mask = state['mask'] > state['p'].seg_thr
|
181 |
frame[:, :, 2] = (mask > 0) * 255 + \
|