Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -298,14 +298,12 @@ def vos_tracking_video(video_state, interactive_state, mask_dropdown):
 
 # inpaint 
 def inpaint_video(video_state, *_args):
-    operation_log = [("",""), ("Inpainting started in safe high-quality mode.","Normal")]
+    operation_log = [("",""), ("Inpainting started in smooth-overlap mode.","Normal")]
 
-    # Получаем кадры и маски
     frames = np.asarray(video_state["origin_images"])
     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 +313,9 @@ def inpaint_video(video_state, *_args):
         if i not in inpaint_mask_numbers:
             inpaint_masks[inpaint_masks == i] = 0
 
-    # Фиксированные настройки
-    chunk_size = 20
-    padding = 30  # по 5 кадров с каждой стороны
+    chunk_size = 30      # увеличим размер чанка
+    save_size = 20       # сколько сохраняем из каждого чанка (уникальное)
+    step = save_size     # шаг между чанками
     fixed_resize_ratio = 1.0
     fixed_dilate_radius = 4
     fixed_raft_iter = 20
@@ -327,17 +325,12 @@ def inpaint_video(video_state, *_args):
     total_len = len(frames)
     inpainted_all = []
 
-    for start in range(0, total_len, chunk_size):
-        # границы с 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)
+    for start in range(0, total_len, step):
+        end = min(start + chunk_size, total_len)
+        chunk_frames = frames[start:end]
+        chunk_masks = inpaint_masks[start:end]
 
-        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]
+        print(f"Inpainting chunk {start}:{end}")
 
         chunk_result = model.baseinpainter.inpaint(
             chunk_frames,
@@ -345,14 +338,20 @@ def inpaint_video(video_state, *_args):
             ratio=fixed_resize_ratio,
             dilate_radius=fixed_dilate_radius,
             raft_iter=fixed_raft_iter,
-            subvideo_length=chunk_size + 2 * padding,
+            subvideo_length=chunk_size,
             neighbor_length=fixed_neighbor_length,
             ref_stride=fixed_ref_stride
         )
 
-        # обрезаем паддинг, сохраняем только нужный фрагмент
-        chunk_central = chunk_result[core_start:core_end]
-        inpainted_all.extend(chunk_central)
+        # Сохраняем только уникальные центральные кадры
+        if start == 0:
+            chunk_to_save = chunk_result[:save_size]
+        elif end == total_len:
+            chunk_to_save = chunk_result[(chunk_size - (end - start)):]
+        else:
+            chunk_to_save = chunk_result[:save_size]
+
+        inpainted_all.extend(chunk_to_save)
 
     output_path = "./result/inpaint/{}".format(video_state["video_name"])
     video_output = generate_video_from_frames(