-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdemo.sh
74 lines (57 loc) · 6.8 KB
/
demo.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# ######## Tuning Gaussian 25 ####################
# python tuning.py --dataset_mode tuning --dataroot /mnt/ssd2/data0/CDIV2K_BSD_clean/4shot --name decoder_ft_gauss25_1shot_128_3layer_v1 \
# --gpu_ids '1' --display_port 8750 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --model finetuning --valid_name 'Kodak24'\
# --backbone_name backbone_S_pretrain_gaussian25_55_v3 --style gauss25 --eval_style gauss25 --decoder_chekpoint decoder_ft_gauss25_1shot_128_3layer_v0 \
# ####### Tuning Gaussian 25 KAN ####################
# python tuning.py --dataset_mode tuning --dataroot /mnt/ssd2/data0/CDIV2K_BSD_clean/4shot --name decoder_ft_gauss25_1shot_4_3layer_kan \
# --gpu_ids '1' --display_port 8750 --batch_size 8 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --model finetuning --valid_name 'Kodak24'\
# --backbone_name backbone_S_pretrain_gaussian25_55_v3 --style gauss25 --eval_style gauss25 --decoder_chekpoint decoder_ft_gauss25_1shot_128_3layer_v0 \
####### Tuning Gaussian 25 KAN_relu ####################
# python tuning.py --dataset_mode tuning --dataroot /mnt/ssd2/data0/CDIV2K_BSD_clean/4shot --config configs/light_relukan.yaml --name decoder_ft_gauss25_1shot_4_3layer_kan_relu \
# --gpu_ids '0' --display_port 8850 --batch_size 8 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --model finetuning --valid_name 'Kodak24'\
# --backbone_name backbone_S_pretrain_gaussian25_55_v3 --style gauss25 --eval_style gauss25 --decoder_chekpoint decoder_ft_gauss25_1shot_128_3layer_v0 \
# ######## Tuning Gaussian 25 MLP ####################
# python tuning.py --dataset_mode tuning --dataroot /mnt/ssd2/data0/CDIV2K_BSD_clean/4shot --config configs/light_mlp.yaml --name decoder_ft_gauss25_1shot_64_3layer_mlp \
# --gpu_ids '1' --display_port 8750 --batch_size 2 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --model finetuning --valid_name 'Kodak24'\
# --backbone_name backbone_S_pretrain_gaussian25_55_v3 --style gauss25 --eval_style gauss25 --decoder_chekpoint decoder_ft_gauss25_1shot_128_3layer_v0 \
# ######## Test Gaussian 25 ####################
# python test.py --dataset_mode test --dataroot_valid /mnt/ssd3/Imagedenoising/ --valid_name 'BSD300' \
# --decoder_chekpoint decoder_ft_gauss25_1shot_128_3layer_v1 --backbone_name backbone_S_pretrain_gaussian25_55_v3 --eval_style gauss25 \
# --name BSD300/decoder_ft_gauss25_1shot_128_3layer_v1 --gpu_ids '0' --model finetuning --epoch best --results_dir /mnt/ssd1/ICCV/Imagedenoising/Results/
# ######### Tuning Gaussian 50 ####################
# # python tuning.py --dataset_mode tuning --dataroot /mnt/ssd1/data0/CDIV2K_BSD_clean/4shot --name decoder_ft_gauss50_1shot_128_3layer_v2 \
# # --gpu_ids '1' --display_port 8800 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --model finetuning --valid_name 'Kodak24'\
# # --backbone_name backbone_S_pretrain_gaussian25_55_v3 --style gauss50 --eval_style gauss50 --decoder_chekpoint decoder_ft_gauss50_1shot_128_3layer_v1 \
# ########## Test Gaussian 50 ####################
# python test.py --dataset_mode test --dataroot_valid /mnt/ssd3/Imagedenoising/ \
# --decoder_chekpoint decoder_ft_gauss50_1shot_128_3layer_v2 --backbone_name backbone_S_pretrain_gaussian25_55_v3 --eval_style gauss50 \
# --name BSD300/decoder_ft_gauss50_1shot_128_3layer_v2 --gpu_ids '0' --model finetuning --epoch best --results_dir /mnt/ssd1/ICCV/Imagedenoising/Results/
# ######### Gaussian 5_50 ####################
# python tuning.py --dataset_mode tuning --dataroot /mnt/ssd1/data0/CDIV2K_BSD_clean/4shot --name decoder_finetune_gauss50_128_3layer_v0\
# --gpu_ids '1' --display_port 8500 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --config configs/light.yaml --model finetuning --lr 1e-4 --valid_name 'Kodak' \
# --backbone_name backbone_S_pretrain_gaussian5_50 --style gauss50 --eval_style gauss50 --decoder_chekpoint decoder_finetune_gauss5_50_1shot_light_128_3layer_v1_new_backbone
#########Test Gaussian 5_50 ####################
# python test.py --dataset_mode test --dataroot_valid /mnt/ssd3/Imagedenoising/BSD300/test --valid_name 'BSD300' \
# --decoder_chekpoint decoder_finetune_gauss5_50_1shot_light_128_3lalyer_v4 --backbone_name backbone_S_pretrain_denoising_gaussian --eval_style gauss5_50 \
# --name BSD300/decoder_finetune_gauss5_50_1shot_light_128_3lalyer_v4 --gpu_ids '0' --model finetuning --epoch best --results_dir /mnt/ssd1/ICCV/Imagedenoising/Results/
# ######### Poisson 30 kan ####################
python tuning.py --dataset_mode tuning --dataroot /mnt/ssd2/data0/CDIV2K_BSD_clean/4shot --config configs/light_kan.yaml --name decoder_ft_poisson30_8_3layer_kan_v0 \
--gpu_ids '0' --display_port 8600 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --model finetuning --valid_name 'Kodak24'\
--backbone_name backbone_S_pretrain_denoising_poisson25_55_v5 --style poisson30 --eval_style poisson30 --decoder_chekpoint decoder_ft_poisson30_128_3layer_v1 \
# # ######### Poisson 30 mlp ####################
# python tuning.py --dataset_mode tuning --dataroot /mnt/ssd1/data0/CDIV2K_BSD_clean/4shot --config configs/light_relukan.yaml --name decoder_ft_poisson30_128_3layer_v2 \
# --gpu_ids '0' --display_port 8600 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --model finetuning --config configs/light_poisson.yaml --valid_name 'Kodak24'\
# --backbone_name backbone_S_pretrain_denoising_poisson25_55_v5 --style poisson30 --eval_style poisson30 --decoder_chekpoint decoder_ft_poisson30_128_3layer_v1 \
# --estimator_chekpoint estimtaor_poisson25_55_v0
# ######### Poisson 50 ####################
# python tuning.py --dataset_mode tuning --dataroot /mnt/ssd1/data0/CDIV2K_BSD_clean/4shot --name decoder_ft_poisson50_128_3layer_v1 \
# --gpu_ids '0' --display_port 8700 --dataroot_valid /mnt/ssd3/Imagedenoising/Kodak24 --model finetuning --config configs/light_poisson.yaml --valid_name 'Kodak24'\
# --backbone_name backbone_S_pretrain_denoising_poisson25_55_v5 --style poisson50 --eval_style poisson50 --decoder_chekpoint decoder_ft_poisson50_128_3layer_v0 \
# --estimator_chekpoint estimtaor_poisson25_55_v0
######### Test Poisson 50 ####################
# python test.py --dataset_mode test --dataroot_valid /mnt/ssd3/Imagedenoising/ \
# --decoder_chekpoint decoder_ft_poisson30_128_3layer_v1 --backbone_name backbone_S_pretrain_denoising_poisson25_55_v5 --eval_style poisson30 \
# --name BSD300/decoder_ft_poisson30_128_3layer_v1 --gpu_ids '0' --model finetuning --epoch best --results_dir /mnt/ssd1/ICCV/Imagedenoising/Results/
# python test.py --dataset_mode test --dataroot_valid /mnt/ssd3/Imagedenoising/ \
# --decoder_chekpoint decoder_ft_poisson50_128_3layer_v0 --backbone_name backbone_S_pretrain_denoising_poisson25_55_v5 --eval_style poisson50 \
# --name BSD300/decoder_ft_poisson50_128_3layer_v0 --gpu_ids '0' --model finetuning --epoch best --results_dir /mnt/ssd1/ICCV/Imagedenoising/Results/