github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -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

1
samplelib/SampleProcessor.py
View file

@ -89,6 +89,7 @@ class SampleProcessor(object):
resolution = opts.get('resolution', None) resolution = opts.get('resolution', None)
if resolution is None: if resolution is None:
continue continue
if resolution not in params_per_resolution:
params_per_resolution[resolution] = imagelib.gen_warp_params(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,