From c00e9454085ef3e96b98a1ded81aad6a3b95d849 Mon Sep 17 00:00:00 2001
From: wangzhibo <wangzhibo@shsening.com>
Date: 星期四, 31 七月 2025 22:20:08 +0800
Subject: [PATCH] add video matting
---
test/test_face_swap.py | 11 ++++-------
1 files changed, 4 insertions(+), 7 deletions(-)
diff --git a/test/test_face_swap.py b/test/test_face_swap.py
index 7462699..675a059 100644
--- a/test/test_face_swap.py
+++ b/test/test_face_swap.py
@@ -1,7 +1,6 @@
-from auto_media_publisher.utils.face_swap.face_swaper_ffmpeg import FaceSwapWorker
-from auto_media_publisher.utils.face_swap.face_swaper_ffmpeg_stream import FFmpegVideoProcessor
+from auto_media_publisher.modifiers.face_swap.face_swapper_enhancer import FaceSwapperEnhancer
import time
-
+import os
import warnings
warnings.filterwarnings("ignore")
@@ -9,13 +8,11 @@
- processor2 = FFmpegVideoProcessor(enable_enhancer=True,enable_nsfw_predictor=False)
+ processor = FaceSwapperEnhancer()
start_time = time.time()
- processor2.run("./temp/lilizhen.jpg","./temp/buhui.mp4", "./temp/buhui-lilizhen-with-enhancer.mp4", num_workers=4)
-
- #worker.swap_face('./temp/lilizhen.jpg', './temp/g1.mp4', './temp/g1_no_enhancer.mp4',1)
+ processor.swap("./temp/buhui.mp4", "./temp/buhui-lilizhen-with-new.mp4","./temp/lilizhen.jpg")
print(time.time()-start_time)
--
Gitblit v1.9.1