github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-10 23:33:30 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

SampleLoader: fix error

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
README.md
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)|
||| |||

8
samplelib/SampleLoader.py
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:
if data is None:
continue
( face_type, ( face_type,
shape, shape,
landmarks, landmarks,
seg_ie_polys, seg_ie_polys,
xseg_mask, xseg_mask,
eyebrows_expand_mod, eyebrows_expand_mod,
source_filename, source_filename ) = data
) in result:
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),