github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-20 05:23:22 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix of converter when using relighted faces in dst

Browse source
This commit is contained in:
Colombo 2019-11-12 17:37:13 +04:00
commit 2167c5cbcb
4 changed files with 16 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
mainscripts/Converter.py
View file

@ -687,7 +687,13 @@ def main (args, device_args):
io.log_err ("%s is not a dfl image file" % (filepath.name) )
continue
source_filename_stem = Path( dflimg.get_source_filename() ).stem
source_filename = dflimg.get_source_filename()
if source_filename is None or source_filename == "_":
continue
source_filename = Path(source_filename)
source_filename_stem = source_filename.stem
if source_filename_stem not in alignments.keys():
alignments[ source_filename_stem ] = []

4
mainscripts/FacesetRelighter.py
View file

@ -235,7 +235,9 @@ def relight(input_dir, lighten=None, random_one=None):
i += 1
cv2_imwrite (relighted_filepath, relighted_img )
dflimg.embed_and_set (relighted_filepath, source_filename="_", relighted=True )
dflimg.remove_source_filename()
dflimg.embed_and_set (relighted_filepath, relighted=True )
except:
io.log_err (f"Exception occured while processing file {filepath.name}. Error: {traceback.format_exc()}")