github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix SampleProcessor

Browse source
This commit is contained in:
iperov 2021-08-11 10:06:18 +04:00
parent 4581d48ff0
commit c8e6f23a31
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
samplelib/SampleProcessor.py
View file

@ -84,19 +84,20 @@ class SampleProcessor(object):
LandmarksProcessor.draw_landmarks (sample_bgr, sample_landmarks, (0, 1, 0)) LandmarksProcessor.draw_landmarks (sample_bgr, sample_landmarks, (0, 1, 0))
params_per_resolution = {} params_per_resolution = {}
warp_rnd_state = np.random.RandomState (sample_rnd_seed-1) warp_rnd_state = np.random.RandomState (sample_rnd_seed-1)
for opts in output_sample_types: for opts in output_sample_types:
resolution = opts.get('resolution', None) resolution = opts.get('resolution', None)
if resolution is None: if resolution is None:
continue continue
params_per_resolution[resolution] = imagelib.gen_warp_params(resolution, if resolution not in params_per_resolution:
params_per_resolution[resolution] = imagelib.gen_warp_params(resolution,
sample_process_options.random_flip, sample_process_options.random_flip,
rotation_range=sample_process_options.rotation_range, rotation_range=sample_process_options.rotation_range,
scale_range=sample_process_options.scale_range, scale_range=sample_process_options.scale_range,
tx_range=sample_process_options.tx_range, tx_range=sample_process_options.tx_range,
ty_range=sample_process_options.ty_range, ty_range=sample_process_options.ty_range,
rnd_state=warp_rnd_state) rnd_state=warp_rnd_state)
outputs_sample = [] outputs_sample = []
for opts in output_sample_types: for opts in output_sample_types:
sample_type = opts.get('sample_type', SPST.NONE) sample_type = opts.get('sample_type', SPST.NONE)