JustAGeek commited on
Commit
fd9985e
·
verified ·
1 Parent(s): 9085c03

Delete duplicate/old file: \/params_storage/params_fig10b_single_gpu.py

Browse files
///params_storage/params_fig10b_single_gpu.py DELETED
@@ -1,178 +0,0 @@
1
- # params_fig10b_single_gpu.py
2
- # ===========================
3
- # Reproduces Fig 10b from the paper:
4
- # Complex environment - Jacobs Hall, July 28 + July 28_2, 4 APs
5
- # Target: median ~64 cm, 90th ~160 cm
6
- #
7
- # Adapted for single GPU (RTX 3060 6GB) with batch_size=8.
8
- # To use: copy this file to ../params.py
9
- # cp params_storage/params_fig10b_single_gpu.py params.py
10
- # Then run: python train_and_test.py
11
- #
12
- # Required data files in ./data/:
13
- # dataset_jacobs_July28.mat
14
- # dataset_fov_train_jacobs_July28_2.mat
15
- # dataset_non_fov_train_jacobs_July28_2.mat
16
- # dataset_fov_test_jacobs_July28_2.mat
17
- # dataset_non_fov_test_jacobs_July28_2.mat
18
- # (Generate these with: python prepare_data.py)
19
-
20
- from easydict import EasyDict as edict
21
- import time
22
- from os.path import join
23
-
24
- opt_exp = edict()
25
-
26
- # ---------- Global Experiment params ----------
27
- opt_exp.isTrain = True
28
- opt_exp.continue_train = False
29
- opt_exp.starting_epoch_count = 0
30
- opt_exp.n_epochs = 50
31
- opt_exp.gpu_ids = ['0'] # Single GPU
32
- opt_exp.data = "rw_to_rw" # Jacobs Hall complex env
33
- opt_exp.n_decoders = 2 # Encoder + location decoder + consistency decoder
34
-
35
- opt_exp.batch_size = 64 # Larger batch for H100-class GPUs
36
- opt_exp.ds_step_trn = 1
37
- opt_exp.ds_step_tst = 1
38
- opt_exp.weight_decay = 1e-5
39
- opt_exp.confidence = False
40
-
41
- # ------ Experiment save paths ----------
42
- opt_exp.save_name = time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime()) # No colons - Windows safe
43
- opt_exp.checkpoints_dir = join('./runs', opt_exp.save_name)
44
- opt_exp.results_dir = opt_exp.checkpoints_dir
45
- opt_exp.log_dir = opt_exp.checkpoints_dir
46
- opt_exp.load_dir = opt_exp.checkpoints_dir
47
-
48
- # ---------- Encoder params ----------
49
- opt_encoder = edict()
50
- opt_encoder.parent_exp = opt_exp
51
- opt_encoder.batch_size = opt_encoder.parent_exp.batch_size
52
- opt_encoder.ngf = 64
53
- opt_encoder.base_model = 'resnet_encoder'
54
- opt_encoder.net = 'G'
55
- opt_encoder.resnet_blocks = 6
56
- opt_encoder.no_dropout = False
57
- opt_encoder.init_type = 'xavier'
58
- opt_encoder.init_gain = 0.02
59
- opt_encoder.norm = 'instance'
60
- opt_encoder.beta1 = 0.5
61
- opt_encoder.lr = 0.00001
62
- opt_encoder.lr_policy = 'step'
63
- opt_encoder.lr_decay_iters = 50
64
- opt_encoder.lambda_L = 1
65
- opt_encoder.lambda_cross = 1e-5
66
- opt_encoder.lambda_reg = 5e-4
67
- opt_encoder.weight_decay = opt_encoder.parent_exp.weight_decay
68
-
69
- opt_encoder.input_nc = 4 # 4 APs in Jacobs Hall
70
- opt_encoder.output_nc = 1
71
- opt_encoder.save_latest_freq = 5000
72
- opt_encoder.save_epoch_freq = 1
73
- opt_encoder.n_epochs = opt_encoder.parent_exp.n_epochs
74
- opt_encoder.isTrain = True
75
- opt_encoder.continue_train = False
76
- opt_encoder.starting_epoch_count = opt_encoder.parent_exp.starting_epoch_count
77
- opt_encoder.name = 'encoder'
78
- opt_encoder.loss_type = "NoLoss"
79
- opt_encoder.niter = 20
80
- opt_encoder.niter_decay = 100
81
-
82
- opt_encoder.gpu_ids = opt_encoder.parent_exp.gpu_ids
83
- opt_encoder.num_threads = 4
84
- opt_encoder.checkpoints_load_dir = opt_encoder.parent_exp.load_dir
85
- opt_encoder.checkpoints_save_dir = opt_encoder.parent_exp.checkpoints_dir
86
- opt_encoder.results_dir = opt_encoder.parent_exp.results_dir
87
- opt_encoder.log_dir = opt_encoder.parent_exp.log_dir
88
- opt_encoder.max_dataset_size = float("inf")
89
- opt_encoder.verbose = False
90
- opt_encoder.suffix = ''
91
-
92
- # ---------- Location Decoder params ----------
93
- opt_decoder = edict()
94
- opt_decoder.parent_exp = opt_exp
95
- opt_decoder.batch_size = opt_decoder.parent_exp.batch_size
96
- opt_decoder.ngf = 64
97
- opt_decoder.base_model = 'resnet_decoder'
98
- opt_decoder.net = 'G'
99
- opt_decoder.resnet_blocks = 9
100
- opt_decoder.encoder_res_blocks = 6
101
- opt_decoder.no_dropout = False
102
- opt_decoder.init_type = 'xavier'
103
- opt_decoder.init_gain = 0.02
104
- opt_decoder.norm = 'instance'
105
- opt_decoder.beta1 = 0.5
106
- opt_decoder.lr = opt_encoder.lr
107
- opt_decoder.lr_policy = 'step'
108
- opt_decoder.lr_decay_iters = 20
109
- opt_decoder.lambda_L = 1
110
- opt_decoder.lambda_cross = 1e-5
111
- opt_decoder.lambda_reg = 5e-4
112
- opt_decoder.weight_decay = opt_decoder.parent_exp.weight_decay
113
-
114
- opt_decoder.input_nc = 4 # 4 APs
115
- opt_decoder.output_nc = 1 # 1-channel location heatmap
116
- opt_decoder.save_latest_freq = 5000
117
- opt_decoder.save_epoch_freq = 1
118
- opt_decoder.n_epochs = opt_decoder.parent_exp.n_epochs
119
- opt_decoder.isTrain = True
120
- opt_decoder.continue_train = False
121
- opt_decoder.starting_epoch_count = opt_decoder.parent_exp.starting_epoch_count
122
- opt_decoder.name = 'decoder'
123
- opt_decoder.loss_type = "L2_sumL1"
124
- opt_decoder.niter = 20
125
- opt_decoder.niter_decay = 100
126
-
127
- opt_decoder.gpu_ids = opt_decoder.parent_exp.gpu_ids
128
- opt_decoder.num_threads = 4
129
- opt_decoder.checkpoints_load_dir = opt_decoder.parent_exp.load_dir
130
- opt_decoder.checkpoints_save_dir = opt_decoder.parent_exp.checkpoints_dir
131
- opt_decoder.results_dir = opt_decoder.parent_exp.results_dir
132
- opt_decoder.log_dir = opt_decoder.parent_exp.log_dir
133
- opt_decoder.verbose = False
134
- opt_decoder.suffix = ''
135
-
136
- # ---------- Consistency (Offset) Decoder params ----------
137
- opt_offset_decoder = edict()
138
- opt_offset_decoder.parent_exp = opt_exp
139
- opt_offset_decoder.batch_size = opt_offset_decoder.parent_exp.batch_size
140
- opt_offset_decoder.ngf = 64
141
- opt_offset_decoder.base_model = 'resnet_decoder'
142
- opt_offset_decoder.net = 'G'
143
- opt_offset_decoder.resnet_blocks = 12
144
- opt_offset_decoder.encoder_res_blocks = 6
145
- opt_offset_decoder.no_dropout = False
146
- opt_offset_decoder.init_type = 'xavier'
147
- opt_offset_decoder.init_gain = 0.02
148
- opt_offset_decoder.norm = 'instance'
149
- opt_offset_decoder.beta1 = 0.5
150
- opt_offset_decoder.lr = opt_encoder.lr
151
- opt_offset_decoder.lr_policy = 'step'
152
- opt_offset_decoder.lr_decay_iters = 50
153
- opt_offset_decoder.lambda_L = 1
154
- opt_offset_decoder.lambda_cross = 0
155
- opt_offset_decoder.lambda_reg = 0
156
- opt_offset_decoder.weight_decay = opt_offset_decoder.parent_exp.weight_decay
157
-
158
- opt_offset_decoder.input_nc = 4 # 4 APs
159
- opt_offset_decoder.output_nc = 4 # 4-channel consistency output
160
- opt_offset_decoder.save_latest_freq = 5000
161
- opt_offset_decoder.save_epoch_freq = 1
162
- opt_offset_decoder.n_epochs = opt_offset_decoder.parent_exp.n_epochs
163
- opt_offset_decoder.isTrain = True
164
- opt_offset_decoder.continue_train = False
165
- opt_offset_decoder.starting_epoch_count = opt_offset_decoder.parent_exp.starting_epoch_count
166
- opt_offset_decoder.name = 'offset_decoder'
167
- opt_offset_decoder.loss_type = "L2_offset_loss"
168
- opt_offset_decoder.niter = 20
169
- opt_offset_decoder.niter_decay = 100
170
-
171
- opt_offset_decoder.gpu_ids = opt_offset_decoder.parent_exp.gpu_ids
172
- opt_offset_decoder.num_threads = 4
173
- opt_offset_decoder.checkpoints_load_dir = opt_offset_decoder.parent_exp.load_dir
174
- opt_offset_decoder.checkpoints_save_dir = opt_offset_decoder.parent_exp.checkpoints_dir
175
- opt_offset_decoder.results_dir = opt_offset_decoder.parent_exp.results_dir
176
- opt_offset_decoder.log_dir = opt_offset_decoder.parent_exp.log_dir
177
- opt_offset_decoder.verbose = False
178
- opt_offset_decoder.suffix = ''