SampleLoader: fix error

2025-07-06 04:52:13 -07:00 · 2020-04-15 16:55:51 +04:00 · 2020-04-15 16:55:51 +04:00 · f8580928ed
commit f8580928ed
parent c42b1b2124
2 changed files with 13 additions and 11 deletions
--- a/samplelib/SampleLoader.py
+++ b/samplelib/SampleLoader.py
@ -74,15 +74,17 @@ class SampleLoader:
        result = FaceSamplesLoaderSubprocessor(image_paths).run()
        sample_list = []

-        for filename, \
-                ( face_type,
-                  shape,
-                  landmarks,
-                  seg_ie_polys,
-                  xseg_mask,
-                  eyebrows_expand_mod,
-                  source_filename,
-                ) in result:
+        for filename, data in result:
+            if data is None:
+                continue
+            ( face_type,
+              shape,
+              landmarks,
+              seg_ie_polys,
+              xseg_mask,
+              eyebrows_expand_mod,
+              source_filename ) = data
+              
            sample_list.append( Sample(filename=filename,
                                        sample_type=SampleType.FACE,
                                        face_type=FaceType.fromString (face_type),