github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

6
DFLIMG/DFLJPG.py
View file

@ -196,10 +196,14 @@ class DFLJPG(object):
else: else:
io.log_err("Unable to encode fanseg_mask for %s" % (filename) ) io.log_err("Unable to encode fanseg_mask for %s" % (filename) )
fanseg_mask = None 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, DFLJPG.embed_dfldict (filename, {'face_type': face_type,
'landmarks': landmarks, 'landmarks': landmarks,
'ie_polys' : ie_polys.dump() if ie_polys is not None else None, 'ie_polys' : ie_polys,
'source_filename': source_filename, 'source_filename': source_filename,
'source_rect': source_rect, 'source_rect': source_rect,
'source_landmarks': source_landmarks, 'source_landmarks': source_landmarks,

8
DFLIMG/DFLPNG.py
View file

@ -312,10 +312,14 @@ class DFLPNG(object):
else: else:
io.log_err("Unable to encode fanseg_mask for %s" % (filename) ) io.log_err("Unable to encode fanseg_mask for %s" % (filename) )
fanseg_mask = None 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, DFLPNG.embed_dfldict (filename, {'face_type': face_type,
'landmarks': landmarks, 'landmarks': landmarks,
'ie_polys' : ie_polys.dump() if ie_polys is not None else None, 'ie_polys' : ie_polys,
'source_filename': source_filename, 'source_filename': source_filename,
'source_rect': source_rect, 'source_rect': source_rect,
'source_landmarks': source_landmarks, 'source_landmarks': source_landmarks,