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| ||bitcoin:bc1qkhh7h0gwwhxgg6h6gpllfgstkd645fefrd5s6z|
|Alipay 捐款|![](doc/Alipay_donation.jpg)| |Alipay 捐款|![](doc/Alipay_donation.jpg)|
||| |||
|Last donations|25$ ( Maria Grasmeder )| |Last donations|10$ ( Mélio Mascort )|
||25$ ( Maria Grasmeder )|
||10$ ( Amien Phillips )| ||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)| |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() result = FaceSamplesLoaderSubprocessor(image_paths).run()
sample_list = [] sample_list = []
for filename, \ for filename, data in result:
( face_type, if data is None:
shape, continue
landmarks, ( face_type,
seg_ie_polys, shape,
xseg_mask, landmarks,
eyebrows_expand_mod, seg_ie_polys,
source_filename, xseg_mask,
) in result: eyebrows_expand_mod,
source_filename ) = data
sample_list.append( Sample(filename=filename, sample_list.append( Sample(filename=filename,
sample_type=SampleType.FACE, sample_type=SampleType.FACE,
face_type=FaceType.fromString (face_type), face_type=FaceType.fromString (face_type),