fix bugs

7a5f8d70 · zycXD · c8c37dbd · 7a5f8d70
Commit 7a5f8d70 authored Sep 11, 2023 by zycXD
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 7 deletions

inference_face.py inference_face.py +8 -7

No files found.
--- a/inference_face.py
+++ b/inference_face.py
@@ -174,6 +174,7 @@ def main() -> None:
                save_restore_path = os.path.join(parent_path, 'restored_faces', save_face_name)
                Image.fromarray(restored_face).save(save_restore_path)

+            if not args.has_aligned:
                # remove padding
                restored_img = restored_img[:lq_resized.height, :lq_resized.width, :]
                # save restored image