CVRPDataset commited on
Commit
bdbfcee
·
verified ·
1 Parent(s): 3e7894f

Delete model_configs/CVRP_deeplabv3plus.py

Browse files
Files changed (1) hide show
  1. model_configs/CVRP_deeplabv3plus.py +0 -303
model_configs/CVRP_deeplabv3plus.py DELETED
@@ -1,303 +0,0 @@
1
- crop_size = (
2
- 512,
3
- 512,
4
- )
5
- data_preprocessor = dict(
6
- bgr_to_rgb=True,
7
- mean=[
8
- 123.675,
9
- 116.28,
10
- 103.53,
11
- ],
12
- pad_val=0,
13
- seg_pad_val=255,
14
- size=(
15
- 512,
16
- 512,
17
- ),
18
- std=[
19
- 58.395,
20
- 57.12,
21
- 57.375,
22
- ],
23
- type='SegDataPreProcessor')
24
- data_root = 'CVRPDataset/'
25
- dataset_type = 'CVRPDataset'
26
- default_hooks = dict(
27
- checkpoint=dict(
28
- by_epoch=False,
29
- interval=2500,
30
- max_keep_ckpts=1,
31
- save_best='mIoU',
32
- type='CheckpointHook'),
33
- logger=dict(interval=100, log_metric_by_epoch=False, type='LoggerHook'),
34
- param_scheduler=dict(type='ParamSchedulerHook'),
35
- sampler_seed=dict(type='DistSamplerSeedHook'),
36
- timer=dict(type='IterTimerHook'),
37
- visualization=dict(type='SegVisualizationHook'))
38
- default_scope = 'mmseg'
39
- env_cfg = dict(
40
- cudnn_benchmark=True,
41
- dist_cfg=dict(backend='nccl'),
42
- mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0))
43
- img_ratios = [
44
- 0.5,
45
- 0.75,
46
- 1.0,
47
- 1.25,
48
- 1.5,
49
- 1.75,
50
- ]
51
- load_from = None
52
- log_level = 'INFO'
53
- log_processor = dict(by_epoch=False)
54
- model = dict(
55
- auxiliary_head=dict(
56
- align_corners=False,
57
- channels=256,
58
- concat_input=False,
59
- dropout_ratio=0.1,
60
- in_channels=1024,
61
- in_index=2,
62
- loss_decode=dict(
63
- loss_weight=0.4, type='CrossEntropyLoss', use_sigmoid=False),
64
- norm_cfg=dict(requires_grad=True, type='BN'),
65
- num_classes=2,
66
- num_convs=1,
67
- type='FCNHead'),
68
- backbone=dict(
69
- contract_dilation=True,
70
- depth=101,
71
- dilations=(
72
- 1,
73
- 1,
74
- 2,
75
- 4,
76
- ),
77
- norm_cfg=dict(requires_grad=True, type='BN'),
78
- norm_eval=False,
79
- num_stages=4,
80
- out_indices=(
81
- 0,
82
- 1,
83
- 2,
84
- 3,
85
- ),
86
- strides=(
87
- 1,
88
- 2,
89
- 1,
90
- 1,
91
- ),
92
- style='pytorch',
93
- type='ResNetV1c'),
94
- data_preprocessor=dict(
95
- bgr_to_rgb=True,
96
- mean=[
97
- 123.675,
98
- 116.28,
99
- 103.53,
100
- ],
101
- pad_val=0,
102
- seg_pad_val=255,
103
- size=(
104
- 512,
105
- 512,
106
- ),
107
- std=[
108
- 58.395,
109
- 57.12,
110
- 57.375,
111
- ],
112
- type='SegDataPreProcessor'),
113
- decode_head=dict(
114
- align_corners=False,
115
- c1_channels=48,
116
- c1_in_channels=256,
117
- channels=512,
118
- dilations=(
119
- 1,
120
- 12,
121
- 24,
122
- 36,
123
- ),
124
- dropout_ratio=0.1,
125
- in_channels=2048,
126
- in_index=3,
127
- loss_decode=dict(
128
- loss_weight=1.0, type='CrossEntropyLoss', use_sigmoid=False),
129
- norm_cfg=dict(requires_grad=True, type='BN'),
130
- num_classes=2,
131
- type='DepthwiseSeparableASPPHead'),
132
- pretrained='open-mmlab://resnet101_v1c',
133
- test_cfg=dict(mode='whole'),
134
- train_cfg=dict(),
135
- type='EncoderDecoder')
136
- norm_cfg = dict(requires_grad=True, type='BN')
137
- optim_wrapper = dict(
138
- clip_grad=None,
139
- optimizer=dict(lr=0.01, momentum=0.9, type='SGD', weight_decay=0.0005),
140
- type='OptimWrapper')
141
- optimizer = dict(lr=0.01, momentum=0.9, type='SGD', weight_decay=0.0005)
142
- param_scheduler = [
143
- dict(
144
- begin=0,
145
- by_epoch=False,
146
- end=160000,
147
- eta_min=0.0001,
148
- power=0.9,
149
- type='PolyLR'),
150
- ]
151
- randomness = dict(seed=0)
152
- resume = False
153
- test_cfg = dict(type='TestLoop')
154
- test_dataloader = dict(
155
- batch_size=1,
156
- dataset=dict(
157
- data_prefix=dict(img_path='img_dir/val', seg_map_path='ann_dir/val'),
158
- data_root='CVRPDataset/',
159
- pipeline=[
160
- dict(type='LoadImageFromFile'),
161
- dict(keep_ratio=True, scale=(
162
- 2048,
163
- 1024,
164
- ), type='Resize'),
165
- dict(type='LoadAnnotations'),
166
- dict(type='PackSegInputs'),
167
- ],
168
- type='CVRPDataset'),
169
- num_workers=4,
170
- persistent_workers=True,
171
- sampler=dict(shuffle=False, type='DefaultSampler'))
172
- test_evaluator = dict(
173
- iou_metrics=[
174
- 'mIoU',
175
- 'mDice',
176
- 'mFscore',
177
- ], type='IoUMetric')
178
- test_pipeline = [
179
- dict(type='LoadImageFromFile'),
180
- dict(keep_ratio=True, scale=(
181
- 2048,
182
- 1024,
183
- ), type='Resize'),
184
- dict(type='LoadAnnotations'),
185
- dict(type='PackSegInputs'),
186
- ]
187
- train_cfg = dict(max_iters=20000, type='IterBasedTrainLoop', val_interval=500)
188
- train_dataloader = dict(
189
- batch_size=4,
190
- dataset=dict(
191
- data_prefix=dict(
192
- img_path='img_dir/train', seg_map_path='ann_dir/train'),
193
- data_root='CVRPDataset/',
194
- pipeline=[
195
- dict(type='LoadImageFromFile'),
196
- dict(type='LoadAnnotations'),
197
- dict(
198
- keep_ratio=True,
199
- ratio_range=(
200
- 0.5,
201
- 2.0,
202
- ),
203
- scale=(
204
- 2048,
205
- 1024,
206
- ),
207
- type='RandomResize'),
208
- dict(
209
- cat_max_ratio=0.75, crop_size=(
210
- 512,
211
- 512,
212
- ), type='RandomCrop'),
213
- dict(prob=0.5, type='RandomFlip'),
214
- dict(type='PhotoMetricDistortion'),
215
- dict(type='PackSegInputs'),
216
- ],
217
- type='CVRPDataset'),
218
- num_workers=2,
219
- persistent_workers=True,
220
- sampler=dict(shuffle=True, type='InfiniteSampler'))
221
- train_pipeline = [
222
- dict(type='LoadImageFromFile'),
223
- dict(type='LoadAnnotations'),
224
- dict(
225
- keep_ratio=True,
226
- ratio_range=(
227
- 0.5,
228
- 2.0,
229
- ),
230
- scale=(
231
- 2048,
232
- 1024,
233
- ),
234
- type='RandomResize'),
235
- dict(cat_max_ratio=0.75, crop_size=(
236
- 512,
237
- 512,
238
- ), type='RandomCrop'),
239
- dict(prob=0.5, type='RandomFlip'),
240
- dict(type='PhotoMetricDistortion'),
241
- dict(type='PackSegInputs'),
242
- ]
243
- tta_model = dict(type='SegTTAModel')
244
- tta_pipeline = [
245
- dict(file_client_args=dict(backend='disk'), type='LoadImageFromFile'),
246
- dict(
247
- transforms=[
248
- [
249
- dict(keep_ratio=True, scale_factor=0.5, type='Resize'),
250
- dict(keep_ratio=True, scale_factor=0.75, type='Resize'),
251
- dict(keep_ratio=True, scale_factor=1.0, type='Resize'),
252
- dict(keep_ratio=True, scale_factor=1.25, type='Resize'),
253
- dict(keep_ratio=True, scale_factor=1.5, type='Resize'),
254
- dict(keep_ratio=True, scale_factor=1.75, type='Resize'),
255
- ],
256
- [
257
- dict(direction='horizontal', prob=0.0, type='RandomFlip'),
258
- dict(direction='horizontal', prob=1.0, type='RandomFlip'),
259
- ],
260
- [
261
- dict(type='LoadAnnotations'),
262
- ],
263
- [
264
- dict(type='PackSegInputs'),
265
- ],
266
- ],
267
- type='TestTimeAug'),
268
- ]
269
- val_cfg = dict(type='ValLoop')
270
- val_dataloader = dict(
271
- batch_size=1,
272
- dataset=dict(
273
- data_prefix=dict(img_path='img_dir/val', seg_map_path='ann_dir/val'),
274
- data_root='CVRPDataset/',
275
- pipeline=[
276
- dict(type='LoadImageFromFile'),
277
- dict(keep_ratio=True, scale=(
278
- 2048,
279
- 1024,
280
- ), type='Resize'),
281
- dict(type='LoadAnnotations'),
282
- dict(type='PackSegInputs'),
283
- ],
284
- type='CVRPDataset'),
285
- num_workers=4,
286
- persistent_workers=True,
287
- sampler=dict(shuffle=False, type='DefaultSampler'))
288
- val_evaluator = dict(
289
- iou_metrics=[
290
- 'mIoU',
291
- 'mDice',
292
- 'mFscore',
293
- ], type='IoUMetric')
294
- vis_backends = [
295
- dict(type='LocalVisBackend'),
296
- ]
297
- visualizer = dict(
298
- name='visualizer',
299
- type='SegLocalVisualizer',
300
- vis_backends=[
301
- dict(type='LocalVisBackend'),
302
- ])
303
- work_dir = './work_dirs/CVRPDataset_DeepLabV3plus'