goryhon commited on
Commit
2530422
·
verified ·
1 Parent(s): fd071b2

Update web-demos/hugging_face/app.py

Browse files
Files changed (1) hide show
  1. web-demos/hugging_face/app.py +3 -1
web-demos/hugging_face/app.py CHANGED
@@ -272,8 +272,10 @@ def vos_tracking_video(video_state, interactive_state, mask_dropdown):
272
 
273
  # Генерация ч/б видео-маски (имитация альфа-канала)
274
  bw_mask_frames = []
 
275
  for mask in video_state["masks"]:
276
- binary_mask = np.where(mask > 0, 255, 0).astype(np.uint8)
 
277
  bw_frame = np.stack([binary_mask]*3, axis=-1) # RGB ч/б
278
  bw_mask_frames.append(bw_frame)
279
 
 
272
 
273
  # Генерация ч/б видео-маски (имитация альфа-канала)
274
  bw_mask_frames = []
275
+ orig_h, orig_w = video_state["origin_images"][0].shape[:2]
276
  for mask in video_state["masks"]:
277
+ mask_up = cv2.resize(mask.astype(np.uint8), (orig_w, orig_h), interpolation=cv2.INTER_NEAREST)
278
+ binary_mask = np.where(mask_up > 0, 255, 0).astype(np.uint8)
279
  bw_frame = np.stack([binary_mask]*3, axis=-1) # RGB ч/б
280
  bw_mask_frames.append(bw_frame)
281