Add -yuv420 option to tools/benchmark.lua
This commit is contained in:
parent
9d3e1a241e
commit
b89833daa3
|
@ -39,6 +39,7 @@ cmd:option("-tta_level", 8, 'tta level')
|
||||||
cmd:option("-crop_size", 128, 'patch size per process')
|
cmd:option("-crop_size", 128, 'patch size per process')
|
||||||
cmd:option("-batch_size", 1, 'batch_size')
|
cmd:option("-batch_size", 1, 'batch_size')
|
||||||
cmd:option("-force_cudnn", 0, 'use cuDNN backend')
|
cmd:option("-force_cudnn", 0, 'use cuDNN backend')
|
||||||
|
cmd:option("-yuv420", 0, 'use yuv420 jpeg')
|
||||||
|
|
||||||
local function to_bool(settings, name)
|
local function to_bool(settings, name)
|
||||||
if settings[name] == 1 then
|
if settings[name] == 1 then
|
||||||
|
@ -54,6 +55,7 @@ if cudnn then
|
||||||
cudnn.benchmark = true
|
cudnn.benchmark = true
|
||||||
end
|
end
|
||||||
to_bool(opt, "force_cudnn")
|
to_bool(opt, "force_cudnn")
|
||||||
|
to_bool(opt, "yuv420")
|
||||||
to_bool(opt, "save_all")
|
to_bool(opt, "save_all")
|
||||||
to_bool(opt, "tta")
|
to_bool(opt, "tta")
|
||||||
if opt.save_all then
|
if opt.save_all then
|
||||||
|
@ -115,7 +117,11 @@ local function transform_jpeg(x, opt)
|
||||||
for i = 1, opt.jpeg_times do
|
for i = 1, opt.jpeg_times do
|
||||||
jpeg = gm.Image(x, "RGB", "DHW")
|
jpeg = gm.Image(x, "RGB", "DHW")
|
||||||
jpeg:format("jpeg")
|
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)
|
blob, len = jpeg:toBlob(opt.jpeg_quality - (i - 1) * opt.jpeg_quality_down)
|
||||||
jpeg:fromBlob(blob, len)
|
jpeg:fromBlob(blob, len)
|
||||||
x = jpeg:toTensor("byte", "RGB", "DHW")
|
x = jpeg:toTensor("byte", "RGB", "DHW")
|
||||||
|
@ -143,7 +149,11 @@ local function transform_scale_jpeg(x, opt)
|
||||||
for i = 1, opt.jpeg_times do
|
for i = 1, opt.jpeg_times do
|
||||||
jpeg = gm.Image(x, "RGB", "DHW")
|
jpeg = gm.Image(x, "RGB", "DHW")
|
||||||
jpeg:format("jpeg")
|
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)
|
blob, len = jpeg:toBlob(opt.jpeg_quality - (i - 1) * opt.jpeg_quality_down)
|
||||||
jpeg:fromBlob(blob, len)
|
jpeg:fromBlob(blob, len)
|
||||||
x = jpeg:toTensor("byte", "RGB", "DHW")
|
x = jpeg:toTensor("byte", "RGB", "DHW")
|
||||||
|
|
Loading…
Reference in a new issue