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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|