Commit 2f6e55a9 authored by Masahide Okada's avatar Masahide Okada
Browse files

modified to trim padding out from aligned faces

parent 904f15a4
......@@ -173,6 +173,8 @@ def main() -> None:
# save restored face
if args.has_aligned:
save_face_name = f'{basename}.{img_save_ext}'
# remove padding
restored_face = restored_face[:lq_resized.height, :lq_resized.width, :]
else:
save_face_name = f'{basename}_{idx:02d}.{img_save_ext}'
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