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

fix previews

Browse source
This commit is contained in:
Jeremy Hummel 2021-05-22 22:33:43 -07:00
commit d9560361d8
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/Model_SAEHD/Model.py
View file

@ -163,6 +163,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
self.options['random_noise'] = io.input_bool("Enable random noise added to samples", False, help_message="") self.options['random_noise'] = io.input_bool("Enable random noise added to samples", False, help_message="")
self.options['random_blur'] = io.input_bool("Enable random blur of samples", False, help_message="") self.options['random_blur'] = io.input_bool("Enable random blur of samples", False, help_message="")
self.options['random_jpeg'] = io.input_bool("Enable random jpeg compression of samples", False, help_message="") self.options['random_jpeg'] = io.input_bool("Enable random jpeg compression of samples", False, help_message="")
self.options['gan_version'] = np.clip (io.input_int("GAN version", default_gan_version, add_info="2 or 3", help_message="Choose GAN version (v2: 7/16/2020, v3: 1/3/2021):"), 2, 3) self.options['gan_version'] = np.clip (io.input_int("GAN version", default_gan_version, add_info="2 or 3", help_message="Choose GAN version (v2: 7/16/2020, v3: 1/3/2021):"), 2, 3)
if self.options['gan_version'] == 2: if self.options['gan_version'] == 2:
@ -853,7 +854,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
wt = [] wt = []
for i in range(n_samples): for i in range(n_samples):
ar = SW[i], SS[i], DW[i], DD[i], SD[i] ar = SW[i], SS[i], DW[i], DD[i], SD[i]
st.append ( np.concatenate ( ar, axis=1) ) wt.append ( np.concatenate ( ar, axis=1) )
result += [ ('SAEHD warped', np.concatenate (wt, axis=0 )), ] result += [ ('SAEHD warped', np.concatenate (wt, axis=0 )), ]
st_m = [] st_m = []