38 lines
1.1 KiB
Bash
Executable file
38 lines
1.1 KiB
Bash
Executable file
#!/bin/sh
|
|
set -x
|
|
|
|
benchmark_photo() {
|
|
dir=./benchmarks/${1}/${2}/${3}
|
|
mkdir -p ${dir}
|
|
th tools/benchmark.lua -dir data/${1} -model1\_dir models/${2}/photo -method scale -filter Catrom -color y -range\_bug 1 -tta ${3} -output_dir ${dir} -save_info 1 -show_progress 0 -force_cudnn 1
|
|
}
|
|
run_benchmark_photo() {
|
|
for tta in 0 1
|
|
do
|
|
for dataset in bsd100 urban100
|
|
do
|
|
benchmark_photo ${dataset} vgg_7 ${tta}
|
|
benchmark_photo ${dataset} upconv_7 ${tta}
|
|
benchmark_photo ${dataset} upconv_7l ${tta}
|
|
done
|
|
done
|
|
}
|
|
benchmark_art() {
|
|
dir=./benchmarks/${1}/${2}/${3}/${4}
|
|
mkdir -p ${dir}
|
|
th tools/benchmark.lua -dir data/${1} -model1\_dir models/${2}/art -method scale -filter ${3} -color y -range\_bug 1 -tta ${4} -output_dir ${dir} -save_info 1 -show_progress 0 -force_cudnn 1
|
|
}
|
|
run_benchmark_art() {
|
|
for tta in 0 1
|
|
do
|
|
for filter in Sinc Lanczos Catrom Box
|
|
do
|
|
benchmark_art art_test vgg_7 ${filter} ${tta}
|
|
benchmark_art art_test upconv_7 ${filter} ${tta}
|
|
benchmark_art art_test cunet ${filter} ${tta}
|
|
done
|
|
done
|
|
}
|
|
|
|
run_benchmark_art
|