Spaces:
Paused
Paused
Update web-demos/hugging_face/app.py
Browse files
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 |
-
|
|
|
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 |
|