-
Notifications
You must be signed in to change notification settings - Fork 3
/
args_parser.py
42 lines (35 loc) · 2 KB
/
args_parser.py
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
import argparse
def args_parser():
parser = argparse.ArgumentParser()
parser.add_argument('-arch', type=str, default='SSRNET',
choices=[# these four models are used for ablation experiments
'SpatCNN', 'SpecCNN',
'SpatRNET', 'SpecRNET',
# the proposed method
'SSRNET',
# these five models are used for comparison experiments
'SSFCNN', 'ConSSFCNN',
'TFNet', 'ResTFNet',
'MSDCNN'
])
parser.add_argument('-root', type=str, default='./data')
parser.add_argument('-dataset', type=str, default='Botswana',
choices=['PaviaU', 'Botswana', 'KSC', 'Urban', 'Pavia', 'IndianP', 'Washington'])
parser.add_argument('--scale_ratio', type=float, default=4)
parser.add_argument('--n_bands', type=int, default=0)
parser.add_argument('--n_select_bands', type=int, default=5)
parser.add_argument('--model_path', type=str,
default='./checkpoints/dataset_arch.pkl',
help='path for trained encoder')
parser.add_argument('--train_dir', type=str, default='./data/dataset/train',
help='directory for resized images')
parser.add_argument('--val_dir', type=str, default='./data/dataset/val',
help='directory for resized images')
# learning settingl
parser.add_argument('--n_epochs', type=int, default=10000,
help='end epoch for training')
# rsicd: 3e-4, ucm: 1e-4,
parser.add_argument('--lr', type=float, default=1e-4)
parser.add_argument('--image_size', type=int, default=128)
args = parser.parse_args()
return args