oguzakif commited on
Commit
6e61d6f
·
1 Parent(s): 7f0774a

first frame mask variable name fixed

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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] = (mask > 0) * 255 + \
167
- (mask == 0) * frame[:, :, 2]
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(