oguzakif commited on
Commit
400f928
1 Parent(s): d52b33a

color fix on siammask output

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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
- frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
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 + \