#!/bin/bash

echo Checkpoint: $1
echo Model_Name: $2
echo GPU: $3

CUDA_VISIBLE_DEVICES=$3

mkdir /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/

CUDA_VISIBLE_DEVICES=$3 python PIH_test.py --datadir /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/File_for_testing/ \
                                           -g 0 \
                                           --checkpoints $1 \
                                           --tmp_results /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/results_testing/ \
                                           --bs 1 \

CUDA_VISIBLE_DEVICES=$3 python PIH_test.py --datadir /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/File_for_testing_different_large/ \
                                           -g 0 \
                                           --checkpoints $1 \
                                           --tmp_results /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/results_testing_different/ \
                                           --bs 1 \                                     


CUDA_VISIBLE_DEVICES=$3 python PIH_test.py --datadir /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/File_for_testing_real_large/ \
                                           -g 0 \
                                           --checkpoints $1 \
                                           --tmp_results /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/results_testing_real/ \
                                           --ngt \
                                           --bs 1 \    
                                           
CUDA_VISIBLE_DEVICES=$3 python PIH_test.py --datadir /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/File_for_testing_real_aug/ \
                                           -g 0 \
                                           --checkpoints $1 \
                                           --tmp_results /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/results_testing_real_aug/ \
                                           --ngt \
                                           --bs 1 \   

mkdir /home/kewang/website_michael/results/$2/

cd /home/kewang/website_michael/ttools2-main/ttools2/scripts/

python /home/kewang/website_michael/ttools2-main/ttools2/scripts/image_gallery.py /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/results_testing/ \
                                                                                  /home/kewang/website_michael/results/$2/results_testing/ \
                                                                                  --order original intermediate results gt \
                                                                                  --name $2-results_testing

python /home/kewang/website_michael/ttools2-main/ttools2/scripts/image_gallery.py /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/results_testing_different/ \
                                                                                  /home/kewang/website_michael/results/$2/results_testing_different/ \
                                                                                  --order original intermediate results gt \
                                                                                  --name $2-results_testing_different

python /home/kewang/website_michael/ttools2-main/ttools2/scripts/image_gallery.py /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/results_testing_real/ \
                                                                                  /home/kewang/website_michael/results/$2/results_testing_real/ \
                                                                                  --order original intermediate results\
                                                                                  --name $2-real

python /home/kewang/website_michael/ttools2-main/ttools2/scripts/image_gallery.py /home/kewang/sensei-fs-symlink/users/kewang/projects/data_processing/results_images/$2/results_testing_real_aug/ \
                                                                                  /home/kewang/website_michael/results/$2/results_testing_real_aug/ \
                                                                                  --order original intermediate results\
                                                                                  --name $2-real-aug

# # Network hyperparameters
# device=1
# lr=1e-5
# batch_size=16
# date=202206070
# name=iharmonysimplemodel

# model_name=exp_${date}_batch_size_$((batch_size))_lr_${lr}_${name}

# # Set folder names
# dir_data=/mnt/localssd/Image_Harmonization_Dataset/
# dir_log=/home/kewang/sensei-fs-symlink/users/kewang/projects/PIH/PIH_ResNet/results/$model_name



# CUDA_VISIBLE_DEVICES=$device python PIH_train.py --datadir $dir_data \
#                        -g 0 \
#                        --logdir $dir_log \
#                        --bs $batch_size \
#                        --lr $lr \
#                        --force_train_from_scratch \
#                        --ihd \