CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-25 23:29:36 +12:00
Code Issues Releases Wiki Activity

Add -yuv420 option to tools/benchmark.lua

Browse source
This commit is contained in:
nagadomi 2016-06-18 20:25:47 +09:00
parent 9d3e1a241e
commit b89833daa3
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tools/benchmark.lua
View file

@ -39,6 +39,7 @@ cmd:option("-tta_level", 8, 'tta level')
cmd:option("-crop_size", 128, 'patch size per process')
cmd:option("-batch_size", 1, 'batch_size')
cmd:option("-force_cudnn", 0, 'use cuDNN backend')
cmd:option("-yuv420", 0, 'use yuv420 jpeg')
local function to_bool(settings, name)
if settings[name] == 1 then
@ -54,6 +55,7 @@ if cudnn then
cudnn.benchmark = true
end
to_bool(opt, "force_cudnn")
to_bool(opt, "yuv420")
to_bool(opt, "save_all")
to_bool(opt, "tta")
if opt.save_all then
@ -115,7 +117,11 @@ local function transform_jpeg(x, opt)
for i = 1, opt.jpeg_times do
jpeg = gm.Image(x, "RGB", "DHW")
jpeg:format("jpeg")
jpeg:samplingFactors({1.0, 1.0, 1.0})
if opt.yuv420 then
jpeg:samplingFactors({2.0, 1.0, 1.0})
else
jpeg:samplingFactors({1.0, 1.0, 1.0})
end
blob, len = jpeg:toBlob(opt.jpeg_quality - (i - 1) * opt.jpeg_quality_down)
jpeg:fromBlob(blob, len)
x = jpeg:toTensor("byte", "RGB", "DHW")
@ -143,7 +149,11 @@ local function transform_scale_jpeg(x, opt)
for i = 1, opt.jpeg_times do
jpeg = gm.Image(x, "RGB", "DHW")
jpeg:format("jpeg")
jpeg:samplingFactors({1.0, 1.0, 1.0})
if opt.yuv420 then
jpeg:samplingFactors({2.0, 1.0, 1.0})
else
jpeg:samplingFactors({1.0, 1.0, 1.0})
end
blob, len = jpeg:toBlob(opt.jpeg_quality - (i - 1) * opt.jpeg_quality_down)
jpeg:fromBlob(blob, len)
x = jpeg:toTensor("byte", "RGB", "DHW")