fix faceset enhancer for frames that contain edited mask

iperov · Dec 29, 2019 · 540650c · 540650c
1 parent 28549dc
commit 540650c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/DFLIMG/DFLJPG.py b/DFLIMG/DFLJPG.py
@@ -196,10 +196,14 @@ def embed_data(filename, face_type=None,
             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,

diff --git a/DFLIMG/DFLPNG.py b/DFLIMG/DFLPNG.py
@@ -312,10 +312,14 @@ def embed_data(filename, face_type=None,
             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,