Skip to content

GitLab

Commit 7a5f8d70 authored by zycXD's avatar zycXD
Browse files

fix bugs

parent c8c37dbd
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 7 deletions
inference_face.py
View file @ 7a5f8d70
...@@ -174,6 +174,7 @@ def main() -> None: ...@@ -174,6 +174,7 @@ def main() -> None:
save_restore_path = os.path.join(parent_path, 'restored_faces', save_face_name) save_restore_path = os.path.join(parent_path, 'restored_faces', save_face_name)
Image.fromarray(restored_face).save(save_restore_path) Image.fromarray(restored_face).save(save_restore_path)
if not args.has_aligned:
# remove padding # remove padding
restored_img = restored_img[:lq_resized.height, :lq_resized.width, :] restored_img = restored_img[:lq_resized.height, :lq_resized.width, :]
# save restored image # save restored image
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment