Fix NaN in benchmark script
This commit is contained in:
parent
a938cd5994
commit
d9474702f6
|
@ -72,7 +72,7 @@ local function MSE(x1, x2, color)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
local function PSNR(x1, x2, color)
|
local function PSNR(x1, x2, color)
|
||||||
local mse = MSE(x1, x2, color)
|
local mse = math.max(MSE(x1, x2, color), 1)
|
||||||
return 10 * math.log10((255.0 * 255.0) / mse)
|
return 10 * math.log10((255.0 * 255.0) / mse)
|
||||||
end
|
end
|
||||||
local function transform_jpeg(x, opt)
|
local function transform_jpeg(x, opt)
|
||||||
|
|
Loading…
Reference in a new issue