inf_get_BasicVSR.sh

source $1

start=$2
end=$3
inter=$4
LR_folder=$5
HR_folder=$6
log=$7
pyenv=$8

conda activate ${pyenv}

# Seperate frames for BasicVSR++ inference
for i in $(seq $start $end);
do
    python seperate_image_to_folders.py --img_dir ${inter}/Inter4K_frame/60fps/${LR_folder}/$i --out_dir ${inter}/Inter4K_frame/60fps/${LR_folder}_split/$i --image_per_folder 5
    echo $i
done

# Get BasicVSR++ inference data

for i in $(seq $start $end);
do
    python multiple_inference.py --img_dir ${inter}/Inter4K_frame/60fps/${LR_folder}_split/$i --out_dir ${inter}/Inter4K_frame/60fps/BasicVSRplusplus_VSRx4_${LR_folder}/$i
    echo $i
done

# Get BasicVSR++ inference results

for i in $(seq $start $end);
do
    python custom_inference.py --gt_dir ${inter}/Inter4K_frame/60fps/${HR_folder}/$i --sr_dir ${inter}/Inter4K_frame/60fps/BasicVSRplusplus_VSRx4_${LR_folder}/$i
    cd mAP && python main.py -na -np
    cd ..
    cp mAP/output/output.txt My_output/BasicVSRplusplus_VSRx4_${LR_folder}_${i}.txt

    echo $i
done
python get_average_psnr.py --gt ${inter}/Inter4K_frame/60fps/${HR_folder}/ --sr ${inter}/Inter4K_frame/60fps/BasicVSRplusplus_VSRx4_${LR_folder}/ --txt $log

# Remove temp
rm -rf ${inter}/Inter4K_frame/60fps/${LR_folder}_split

# Remove BasicVSR++ data to save space
# rm -rf ${inter}/Inter4K_frame/60fps/BasicVSRplusplus_VSRx4_${LR_folder}