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 +