소스 검색

Fix NaN in benchmark script

nagadomi 9 년 전
부모
커밋
d9474702f6
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      tools/benchmark.lua

+ 1 - 1
tools/benchmark.lua

@@ -72,7 +72,7 @@ local function MSE(x1, x2, color)
    end
 end
 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)
 end
 local function transform_jpeg(x, opt)