Skip to content

GitLab

Unverified Commit 05b0c15a authored by Ziyan Chen's avatar Ziyan Chen Committed by GitHub
Browse files

Merge pull request #28 from MasahideOkada/trim-padding 

fix padding-removal in aligned face inputs
parents a27df00b 2f6e55a9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
inference_face.py
View file @ 05b0c15a
...@@ -222,6 +222,8 @@ def main() -> None: ...@@ -222,6 +222,8 @@ def main() -> None:
# save restored face # save restored face
if args.has_aligned: if args.has_aligned:
save_face_name = f'{basename}.{img_save_ext}' save_face_name = f'{basename}.{img_save_ext}'
# remove padding
restored_face = restored_face[:lq_resized.height, :lq_resized.width, :]
else: else:
save_face_name = f'{basename}_{idx:02d}.{img_save_ext}' save_face_name = f'{basename}_{idx:02d}.{img_save_ext}'
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)
......
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