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

fix: misnamed arg for tf.image.ssim_multiscale

Browse source
This commit is contained in:
jh 2021-03-12 08:18:56 -08:00
commit 6b78f1cbc3
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/leras/ops/__init__.py
View file

@ -325,7 +325,7 @@ def ms_ssim(img1, img2, resolution, kernel_size=11, k1=0.01, k2=0.03, max_value=
if sum(power_factors) < 1.0: if sum(power_factors) < 1.0:
power_factors = [x/sum(power_factors) for x in power_factors] power_factors = [x/sum(power_factors) for x in power_factors]
ms_ssim_val = tf.image.ssim_multiscale(img1, img2, max_value=max_value, power_factors=power_factors, ms_ssim_val = tf.image.ssim_multiscale(img1, img2, max_val=max_value, power_factors=power_factors,
filter_size=kernel_size, k1=k1, k2=k2) filter_size=kernel_size, k1=k1, k2=k2)
loss = (1.0 - ms_ssim_val) / 2.0 loss = (1.0 - ms_ssim_val) / 2.0