Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
DiffBIR_pytorch
Commits
16bc7657
Commit
16bc7657
authored
Sep 15, 2023
by
zycXD
Browse files
Merge branch 'main' of
https://github.com/XPixelGroup/DiffBIR
into main
parents
e5a6b0a4
05b0c15a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
inference_face.py
inference_face.py
+2
-0
No files found.
inference_face.py
View file @
16bc7657
...
@@ -238,6 +238,8 @@ def main() -> None:
...
@@ -238,6 +238,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
:
02
d
}
.
{
img_save_ext
}
'
save_face_name
=
f
'
{
basename
}
_
{
idx
:
02
d
}
.
{
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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment