CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix faceset enhancer for frames that contain edited mask

Browse Source
This commit is contained in:
Colombo
2019-12-29 18:10:05 +04:00
parent 28549dc153
commit 540650c610
2 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
DFLIMG/DFLJPG.py
+5 -1
View File
@@ -196,10 +196,14 @@ class DFLJPG(object):
else:
io.log_err("Unable to encode fanseg_mask for %s" % (filename) )
fanseg_mask = None
if ie_polys is not None:
if not isinstance(ie_polys, list):
ie_polys = ie_polys.dump()
DFLJPG.embed_dfldict (filename, {'face_type': face_type,
'landmarks': landmarks,
'ie_polys' : ie_polys.dump() if ie_polys is not None else None,
'ie_polys' : ie_polys,
'source_filename': source_filename,
'source_rect': source_rect,
'source_landmarks': source_landmarks,
DFLIMG/DFLPNG.py
+6 -2
View File
@@ -312,10 +312,14 @@ class DFLPNG(object):
else:
io.log_err("Unable to encode fanseg_mask for %s" % (filename) )
fanseg_mask = None
if ie_polys is not None:
if not isinstance(ie_polys, list):
ie_polys = ie_polys.dump()
DFLPNG.embed_dfldict (filename, {'face_type': face_type,
'landmarks': landmarks,
'ie_polys' : ie_polys.dump() if ie_polys is not None else None,
'ie_polys' : ie_polys,
'source_filename': source_filename,
'source_rect': source_rect,
'source_landmarks': source_landmarks,