MendoPower commited on
Commit
6569624
·
verified ·
1 Parent(s): 60d92f2

Delete grounding_dino_swin-t_finetune_8xb2_20e_concrete.py

Browse files
grounding_dino_swin-t_finetune_8xb2_20e_concrete.py DELETED
@@ -1,59 +0,0 @@
1
- # _base_ = 'grounding_dino_swin-t_finetune_16xb2_1x_coco.py'
2
- _base_ = './grounding_dino_swin-t_finetune_16xb2_1x_coco.py'
3
-
4
- data_root = 'data/concrete_defect/'
5
- class_name = ('crack', 'corrosion', 'efflorescence', 'pothole', 'spalling')
6
- num_classes = len(class_name)
7
- metainfo = dict(classes=class_name,
8
- palette=[(244, 108, 59), (99, 102, 129), (249, 193, 0), (160, 180, 0), (115, 82, 59)])
9
-
10
- model = dict(bbox_head=dict(num_classes=num_classes))
11
-
12
- train_dataloader = dict(
13
- dataset=dict(
14
- data_root=data_root,
15
- metainfo=metainfo,
16
- ann_file='annotations/trainval.json',
17
- data_prefix=dict(img='images/')))
18
-
19
- val_dataloader = dict(
20
- dataset=dict(
21
- metainfo=metainfo,
22
- data_root=data_root,
23
- ann_file='annotations/test.json',
24
- data_prefix=dict(img='images/')))
25
-
26
- test_dataloader = val_dataloader
27
-
28
- val_evaluator = dict(ann_file=data_root + 'annotations/test.json')
29
- test_evaluator = val_evaluator
30
-
31
- max_epoch = 20
32
-
33
- default_hooks = dict(
34
- checkpoint=dict(interval=1, max_keep_ckpts=1, save_best='auto'),
35
- logger=dict(type='LoggerHook', interval=5))
36
- train_cfg = dict(max_epochs=max_epoch, val_interval=1)
37
-
38
- param_scheduler = [
39
- dict(type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=30),
40
- dict(
41
- type='MultiStepLR',
42
- begin=0,
43
- end=max_epoch,
44
- by_epoch=True,
45
- milestones=[15],
46
- gamma=0.1)
47
- ]
48
-
49
- optim_wrapper = dict(
50
- optimizer=dict(lr=0.00005),
51
- paramwise_cfg=dict(
52
- custom_keys={
53
- 'absolute_pos_embed': dict(decay_mult=0.),
54
- 'backbone': dict(lr_mult=0.1),
55
- 'language_model': dict(lr_mult=0),
56
- }))
57
-
58
- auto_scale_lr = dict(base_batch_size=16)
59
- # auto_scale_lr = dict(base_batch_size=4)