Commit 16bc7657 authored by zycXD's avatar zycXD
Browse files

Merge branch 'main' of https://github.com/XPixelGroup/DiffBIR into main

parents e5a6b0a4 05b0c15a
......@@ -238,6 +238,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