From 56d194a4300610a17cc3c6682aaf63f039e9c4ab Mon Sep 17 00:00:00 2001
From: wangzhibo <wangzhibo@shsening.com>
Date: 星期日, 03 八月 2025 16:11:37 +0800
Subject: [PATCH] add matanyone-better auqliaty than robustvideomatting
---
test/test_video_matting.py | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/test/test_video_matting.py b/test/test_video_matting.py
index 7d9e8f4..6bb5700 100644
--- a/test/test_video_matting.py
+++ b/test/test_video_matting.py
@@ -1,6 +1,6 @@
-from auto_media_publisher.modifiers.video_matting.video_matting_converter import VideoMattingConverter
+from auto_media_publisher.modifiers.video_matting.robust_video_matting_converter import RobustVideoMattingConverter
import time
if __name__ == '__main__':
import argparse
@@ -8,7 +8,7 @@
parser = argparse.ArgumentParser()
parser.add_argument('--input', type=str, required=True)
parser.add_argument('--bg_video', type=str)
- parser.add_argument('--bg_jpg', type=str)
+ parser.add_argument('--bg_image', type=str)
parser.add_argument('--output_video', type=str)
parser.add_argument('--output_alpha', type=str)
parser.add_argument('--output_fore', type=str)
@@ -17,7 +17,7 @@
time_start = time.time()
- converter = VideoMattingConverter()
+ converter = RobustVideoMattingConverter()
print(f"model loaded .. {time.time()-time_start}")
time_start = time.time()
@@ -25,7 +25,7 @@
converter.video_matting(
input=args.input,
bg_video=args.bg_video,
- bg_jpg=args.bg_jpg,
+ bg_image=args.bg_image,
output_video=args.output_video,
output_alpha=args.output_alpha,
output_fore=args.output_fore,
--
Gitblit v1.9.1