wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-08-12 12:10:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

* fix doc style

Browse Source
This commit is contained in:
tboquet
2017-03-19 13:43:50 -04:00
parent c23d3540f3
commit 399fbe82bc
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/losses.py
+6 -4
View File
@@ -8,10 +8,12 @@ class DSSIMObjective():
"""
Difference of Structural Similarity (DSSIM loss function). Clipped between 0 and 0.5
Note : You should add a regularization term like a l2 loss in addition to this one.
:param k1: Parameter of the SSIM (default 0.01)
:param k2: Parameter of the SSIM (default 0.03)
:param kernel_size: Size of the sliding window (default 3)
:param max_value: Max value of the output (default 1.0)
# Arguments
k1: Parameter of the SSIM (default 0.01)
k2: Parameter of the SSIM (default 0.03)
kernel_size: Size of the sliding window (default 3)
max_value: Max value of the output (default 1.0)
"""
self.__name__ = "DSSIMObjective"
self.kernel_size = kernel_size