CalvinBackup/iperov-DeepFaceLab
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ConverterMasked: better lct

Browse Source
This commit is contained in:
iperov committed 2019-02-14 18:43:24 +04:00
1 parent fbf6b5f25e
commit 75eb7468ee
2 files changed
+4 -3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

samples/SampleGeneratorFace.py
+2 -2
View File
@@ -157,10 +157,10 @@ class SampleGeneratorFace(SampleGeneratorBase):
pitch, yaw = LandmarksProcessor.estimate_pitch_yaw (sample.landmarks)
if self.add_pitch:
batches[i_pitch].append (pitch)
batches[i_pitch].append ([pitch])
if self.add_yaw:
batches[i_yaw].append (yaw)
batches[i_yaw].append ([yaw])
break
yield [ np.array(batch) for batch in batches]