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

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 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: