修复bug

2026-05-22 14:17:32 +08:00 · 2023-12-26 19:12:48 +08:00
parent f7ce8013e0
commit 18d57f2a18
3 changed files with 47 additions and 18 deletions
--- a/backend/inpaint/sttn_inpaint.py
+++ b/backend/inpaint/sttn_inpaint.py
@@ -87,6 +87,7 @@ class STTNInpaint:
                    frame[inpaint_area[k][0]:inpaint_area[k][1], :, :] = mask_area * comp + (1 - mask_area) * frame[inpaint_area[k][0]:inpaint_area[k][1], :, :]
                # 将最终帧添加到列表
                inpainted_frames.append(frame)
+                print(f'processing frame, {len(frames_hr) - j} left')
        return inpainted_frames

    @staticmethod