david/DeepFaceLab_Linux
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
mainscripts/Sorter.py
+5 -3
View File
@@ -553,10 +553,12 @@ def sort_final(input_path):
imgs_per_grad = 15 imgs_per_grad = 15
sharpned_imgs_per_grad = imgs_per_grad*10 sharpned_imgs_per_grad = imgs_per_grad*10
grads_space = np.linspace (-255,255,grads)
yaws_sample_list = [None]*grads yaws_sample_list = [None]*grads
for g in tqdm ( range (grads), desc="Sort by yaw" ): for g in tqdm ( range(grads), desc="Sort by yaw" ):
yaw = -grads+1 + g*2 yaw = grads_space[g]
next_yaw = -grads+1 + (g+1)*2 next_yaw = grads_space[g+1] if g < grads-1 else yaw
yaw_samples = [] yaw_samples = []
for img in img_list: for img in img_list: