#!/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