Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -305,7 +305,7 @@ def inpaint_video(video_state, *_args):
     fps = video_state["fps"]
     inpaint_masks = np.asarray(video_state["masks"])
 
-    # Маскировка (если не выбраны вручную)
+    # Маскировка
     mask_dropdown = _args[-1]
     if len(mask_dropdown) == 0:
         mask_dropdown = ["mask_001"]
@@ -315,9 +315,10 @@ def inpaint_video(video_state, *_args):
         if i not in inpaint_mask_numbers:
             inpaint_masks[inpaint_masks == i] = 0
 
-    # Фиксированные безопасные настройки
-    chunk_size = 20              # обрабатываем по 20 кадров
-    fixed_resize_ratio = 1.0     # не уменьшаем разрешение
+    # Фиксированные настройки
+    chunk_size = 20
+    padding = 5  # по 5 кадров с каждой стороны
+    fixed_resize_ratio = 1.0
     fixed_dilate_radius = 4
     fixed_raft_iter = 20
     fixed_neighbor_length = 5
@@ -327,25 +328,32 @@ def inpaint_video(video_state, *_args):
     inpainted_all = []
 
     for start in range(0, total_len, chunk_size):
-        end = min(start + chunk_size, total_len)
-        print(f"Inpainting chunk {start} → {end}")
-        chunk_frames = frames[start:end]
-        chunk_masks = inpaint_masks[start:end]
+        # границы с padding
+        pad_start = max(0, start - padding)
+        pad_end = min(total_len, start + chunk_size + padding)
+        core_start = start - pad_start
+        core_end = core_start + min(chunk_size, total_len - start)
+
+        print(f"Inpainting chunk {start} → {min(start + chunk_size, total_len)} with context [{pad_start}:{pad_end}]")
+
+        chunk_frames = frames[pad_start:pad_end]
+        chunk_masks = inpaint_masks[pad_start:pad_end]
 
-        # Инференс inpainting
         chunk_result = model.baseinpainter.inpaint(
             chunk_frames,
             chunk_masks,
             ratio=fixed_resize_ratio,
             dilate_radius=fixed_dilate_radius,
             raft_iter=fixed_raft_iter,
-            subvideo_length=chunk_size,
+            subvideo_length=chunk_size + 2 * padding,
             neighbor_length=fixed_neighbor_length,
             ref_stride=fixed_ref_stride
         )
-        inpainted_all.extend(chunk_result)
 
-    # Сохраняем с повышенным битрейтом
+        # обрезаем паддинг, сохраняем только нужный фрагмент
+        chunk_central = chunk_result[core_start:core_end]
+        inpainted_all.extend(chunk_central)
+
     output_path = "./result/inpaint/{}".format(video_state["video_name"])
     video_output = generate_video_from_frames(
         inpainted_all,