SampleLoader: fix error

This commit is contained in:
Colombo 2020-04-15 16:55:51 +04:00
parent c42b1b2124
commit f8580928ed
2 changed files with 13 additions and 11 deletions

View file

@ -153,9 +153,9 @@ Unfortunately, there is no "make everything ok" button in DeepFaceLab. You shoul
||bitcoin:bc1qkhh7h0gwwhxgg6h6gpllfgstkd645fefrd5s6z|
|Alipay 捐款|![](doc/Alipay_donation.jpg)|
|||
|Last donations|25$ ( Maria Grasmeder )|
|Last donations|10$ ( Mélio Mascort )|
||25$ ( Maria Grasmeder )|
||10$ ( Amien Phillips )|
||20$ ( Ralph Gathmann )|
|||
|Collect facesets|You can collect faceset of any celebrity that can be used in DeepFaceLab and share it [in the community](https://mrdeepfakes.com/forums/forum-celebrity-facesets)|
|||

View file

@ -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),