-
Notifications
You must be signed in to change notification settings - Fork 20
/
job_dncnn.sh
84 lines (62 loc) · 1.88 KB
/
job_dncnn.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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#!/bin/bash
#SBATCH --gres=gpu:1
#SBATCH --mem=40G
#SBATCH --output=logs/job_output_%j.out
#SBATCH --error=logs/job_error_%j.out
#SBATCH --job-name=dncnn_nf
hostname
whoami
### Training
## Train DnCNN with Gaussian noise
python train_dncnn_noiseflow.py \
--model DnCNN_Gauss \
--train_data '/home/abdo/Downloads/SIDD_Medium_Raw/Data' \
--save_every 20 \
--max_epoch 2000 \
--num_gpus 1
## Train DnCNN with signal-dependent noise using camera noise level functions
#python train_dncnn_noiseflow.py \
# --model DnCNN_CamNLF \
# --train_data '/home/abdo/Downloads/SIDD_Medium_Raw/Data' \
# --save_every 20 \
# --max_epoch 2000 \
# --num_gpus 1
## Train DnCNN with noise generated with Noise Flow
#python train_dncnn_noiseflow.py \
# --model DnCNN_NF \
# --train_data '/home/abdo/Downloads/SIDD_Medium_Raw/Data' \
# --save_every 20 \
# --max_epoch 2000 \
# --num_gpus 1
## Train DnCNN with real noise
#python train_dncnn_noiseflow.py \
# --model DnCNN_Real \
# --train_data '/home/abdo/Downloads/SIDD_Medium_Raw/Data' \
# --save_every 20 \
# --max_epoch 2000 \
# --num_gpus 1
### Testing
#python test_dncnn_noiseflow.py \
# --model_name 'DnCNN_Gauss' \
# --save_result \
# --min_epc 1 \
# --max_epc 2000 \
# --epc_step 20
#python test_dncnn_noiseflow.py \
# --model_name 'DnCNN_CamNLF' \
# --save_result \
# --min_epc 1 \
# --max_epc 2000 \
# --epc_step 20
#python test_dncnn_noiseflow.py \
# --model_name 'DnCNN_NF' \
# --save_result \
# --min_epc 1 \
# --max_epc 2000 \
# --epc_step 20
#python test_dncnn_noiseflow.py \
# --model_name 'DnCNN_Real' \
# --save_result \
# --min_epc 1 \
# --max_epc 2000 \
# --epc_step 20