github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-14 02:37:00 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix scale factor in sample generation. added util --export-faceset-mask

Browse source
This commit is contained in:
iperov 2022-07-23 10:30:22 +04:00
commit 521a23f557
3 changed files with 37 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/imagelib/warp.py
View file

@ -115,7 +115,7 @@ def gen_warp_params (w, flip=False, rotation_range=[-10,10], scale_range=[-0.5,
w = 64
rotation = rnd_state.uniform( rotation_range[0], rotation_range[1] )
scale = rnd_state.uniform(1 +scale_range[0], 1 +scale_range[1])
scale = rnd_state.uniform( 1/(1-scale_range[0]) , 1+scale_range[1] )
tx = rnd_state.uniform( tx_range[0], tx_range[1] )
ty = rnd_state.uniform( ty_range[0], ty_range[1] )
p_flip = flip and rnd_state.randint(10) < 4