CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-02 19:14:30 +12:00
Code Issues Releases Wiki Activity

Fix NaN in PSNRCriterion

Browse source
This commit is contained in:
nagadomi 2016-03-12 08:47:14 +09:00
parent 68c06feffd
commit 5590674783
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib/PSNRCriterion.lua
View file

@ -10,7 +10,7 @@ function PSNRCriterion:updateOutput(input, target)
self.image:clamp(0.0, 1.0)
self.diff:resizeAs(self.image):copy(self.image)
local mse = self.diff:add(-1, target):pow(2):mean()
local mse = math.max(self.diff:add(-1, target):pow(2):mean(), (0.1/255)^2)
self.output = 10 * math.log10(1.0 / mse)
return self.output
end