oguzakif commited on
Commit
f1aa497
1 Parent(s): 739f13f

first frame added to segmented video

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -165,6 +165,8 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
165
  state = siamese_init(
166
  frame, target_pos, target_sz, siammask, cfg['hp'], device=device)
167
  original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
 
 
168
  else:
169
  # track
170
  state = siamese_track(
@@ -177,7 +179,8 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
177
  mask = mask.astype(np.uint8) # convert to an unsigned byte
178
  mask = mask * 255
179
  mask_list.append(mask)
180
- video_writer.write(frame)
 
181
 
182
  f = f + 1
183
 
 
165
  state = siamese_init(
166
  frame, target_pos, target_sz, siammask, cfg['hp'], device=device)
167
  original_list.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
168
+ frame[:, :, 2] = (masked_frame > 0) * 255 + \
169
+ (masked_frame == 0) * frame[:, :, 2]
170
  else:
171
  # track
172
  state = siamese_track(
 
179
  mask = mask.astype(np.uint8) # convert to an unsigned byte
180
  mask = mask * 255
181
  mask_list.append(mask)
182
+
183
+ video_writer.write(frame)
184
 
185
  f = f + 1
186