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

fix sort by final

Browse Source
This commit is contained in:
iperov
2019-01-07 14:57:46 +04:00
parent cce2e8a0ba
commit ff3a208e80
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mainscripts/Sorter.py
View File

@@ -553,10 +553,12 @@ def sort_final(input_path):
imgs_per_grad = 15
sharpned_imgs_per_grad = imgs_per_grad*10
grads_space = np.linspace (-255,255,grads)
yaws_sample_list = [None]*grads
for g in tqdm ( range (grads), desc="Sort by yaw" ):
yaw = -grads+1 + g*2
next_yaw = -grads+1 + (g+1)*2
for g in tqdm ( range(grads), desc="Sort by yaw" ):
yaw = grads_space[g]
next_yaw = grads_space[g+1] if g < grads-1 else yaw
yaw_samples = []
for img in img_list: