Spaces:

ZJYang
/

AniPortrait_official

Running on Zero

zejunyang commited on Apr 9

Commit

9600e7d

•

1 Parent(s): ac336de

last push

Files changed (1) hide show

app.py CHANGED Viewed

@@ -234,7 +234,7 @@ def video2video(ref_img, source_video, size=512, steps=25, length=60, seed=42):
     save_dir.mkdir(exist_ok=True, parents=True)
     ref_image_np = cv2.cvtColor(ref_img, cv2.COLOR_RGB2BGR)
-    # ref_image_np = crop_face(ref_image_np, lmk_extractor)
     if ref_image_np is None:
         return None, Image.fromarray(ref_img)

     save_dir.mkdir(exist_ok=True, parents=True)
     ref_image_np = cv2.cvtColor(ref_img, cv2.COLOR_RGB2BGR)
+    ref_image_np = crop_face(ref_image_np, lmk_extractor)
     if ref_image_np is None:
         return None, Image.fromarray(ref_img)