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

fixed sort by yaw feeder

Browse Source
This commit is contained in:
iperov
2019-01-07 09:40:00 +04:00
parent 3b0b1a7dec
commit 12dd7fcae5
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
samples/SampleLoader.py
View File

@@ -72,9 +72,9 @@ class SampleLoader:
return sample_list
@staticmethod
def upgradeToFaceYawSortedSamples( YAW_RAWS ):
def upgradeToFaceYawSortedSamples( samples ):
lowest_yaw, highest_yaw = -32, +32
lowest_yaw, highest_yaw = -256, +256
gradations = 64
diff_rot_per_grad = abs(highest_yaw-lowest_yaw) / gradations
@@ -85,7 +85,7 @@ class SampleLoader:
next_yaw = lowest_yaw + (i+1)*diff_rot_per_grad
yaw_samples = []
for s in YAW_RAWS:
for s in samples:
s_yaw = s.yaw
if (i == 0 and s_yaw < next_yaw) or \
(i < gradations-1 and s_yaw >= yaw and s_yaw < next_yaw) or \