mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-08-21 22:13:20 -07:00
kernel size fix
This commit is contained in:
parent
1785ecc77d
commit
703c790e67
1 changed files with 1 additions and 1 deletions
|
@ -20,7 +20,7 @@ class MsSsim(nn.LayerBase):
|
||||||
y_true_t = tf.transpose(tf.cast(y_true, tf.float32), [0, 2, 3, 1])
|
y_true_t = tf.transpose(tf.cast(y_true, tf.float32), [0, 2, 3, 1])
|
||||||
y_pred_t = tf.transpose(tf.cast(y_pred, tf.float32), [0, 2, 3, 1])
|
y_pred_t = tf.transpose(tf.cast(y_pred, tf.float32), [0, 2, 3, 1])
|
||||||
|
|
||||||
loss = tf.image.ssim_multiscale(y_true_t, y_pred_t, max_val, power_factors=self.power_factors)
|
loss = tf.image.ssim_multiscale(y_true_t, y_pred_t, max_val, power_factors=self.power_factors, filter_size=self.kernel_size)
|
||||||
return (1.0 - loss) / 2.0
|
return (1.0 - loss) / 2.0
|
||||||
|
|
||||||
nn.MsSsim = MsSsim
|
nn.MsSsim = MsSsim
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue