nagadomi 9 years ago
parent
commit
2b9753157a
1 changed files with 2 additions and 3 deletions
  1. 2 3
      tools/benchmark.lua

+ 2 - 3
tools/benchmark.lua

@@ -32,7 +32,6 @@ if cudnn then
    cudnn.benchmark = false
 end
 
-
 local function MSE(x1, x2)
    return (x1 - x2):pow(2):mean()
 end
@@ -43,11 +42,11 @@ local function YMSE(x1, x2)
 end
 local function PSNR(x1, x2)
    local mse = MSE(x1, x2)
-   return 20 * (math.log(1.0 / math.sqrt(mse)) / math.log(10))
+   return 10 * math.log10(1.0 / mse)
 end
 local function YPSNR(x1, x2)
    local mse = YMSE(x1, x2)
-   return 20 * (math.log(1.0 / math.sqrt(mse)) / math.log(10))
+   return 10 * math.log10(1.0 / mse)
 end
 
 local function transform_jpeg(x)