Chris MyauandClaude Sonnet 4.6
df9d9eb7c0
Wraps the Adobe Research PIH model (CVPR 2023) as ComfyUI nodes. Includes auto-download of the pretrained checkpoint on first use. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
84 lines
5.3 KiB
Bash
84 lines
5.3 KiB
Bash
#!/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 \
|