Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -264,8 +264,16 @@ def vos_tracking_video(video_state, interactive_state, mask_dropdown):
         video_state["masks"][video_state["select_frame_number"]:] = masks
         video_state["logits"][video_state["select_frame_number"]:] = logits
         video_state["painted_images"][video_state["select_frame_number"]:] = painted_images
+        
+# Генерация ч/б видео-маски (имитация альфа-канала)
+        bw_mask_frames = []
+        for mask in video_state["masks"]:
+            binary_mask = np.where(mask > 0, 255, 0).astype(np.uint8)
+            bw_frame = np.stack([binary_mask]*3, axis=-1)  # RGB ч/б
+            bw_mask_frames.append(bw_frame)
 
-    video_output = generate_video_from_frames(video_state["painted_images"], output_path="./result/track/{}".format(video_state["video_name"]), fps=float(fps)) # import video_input to name the output video
+        # Заменяем визуальное видео на ч/б маску
+    video_output = generate_video_from_frames(bw_mask_frames, output_path="./result/track/{}".format(video_state["video_name"]), fps=float(fps))# import video_input to name the output video
     interactive_state["inference_times"] += 1
 
     print("Tracking resolution:", following_frames[0].shape)