From c8bf9a3fb2afe4e441d48a1ffdceb039b5c703cc Mon Sep 17 00:00:00 2001 From: raven Date: Thu, 16 May 2024 03:07:44 +0900 Subject: [PATCH] Fix audio2vid.py to resize properly --- scripts/audio2vid.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/audio2vid.py b/scripts/audio2vid.py index 866ce83..3718c6d 100644 --- a/scripts/audio2vid.py +++ b/scripts/audio2vid.py @@ -147,7 +147,7 @@ def main(): ref_image_pil = Image.open(ref_image_path).convert("RGB") ref_image_np = cv2.cvtColor(np.array(ref_image_pil), cv2.COLOR_RGB2BGR) - ref_image_np = cv2.resize(ref_image_np, (args.H, args.W)) + ref_image_np = cv2.resize(ref_image_np, (args.W, args.H)) face_result = lmk_extractor(ref_image_np) assert face_result is not None, "No face detected." @@ -266,4 +266,4 @@ def main(): if __name__ == "__main__": main() - \ No newline at end of file +