File size: 1,065 Bytes
ce209f5 | 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 | _base_ = '/new-home/buddhiw/CD-Models/model_repos/open-cd/configs/changer/changer_ex_r18_512x512_40k_levircd.py'
dataset_type = 'DSIFN_Dataset'
data_root = r'/new-home/buddhiw/CD-Models/generated_dataset_views/levir_cd_test_as_val/opencd'
crop_size = (256, 256)
train_dataloader = dict(
batch_size=8,
num_workers=4,
dataset=dict(
type=dataset_type,
data_root=data_root,
data_prefix=dict(seg_map_path='train/label', img_path_from='train/A', img_path_to='train/B')))
val_dataloader = dict(
dataset=dict(
type=dataset_type,
data_root=data_root,
data_prefix=dict(seg_map_path='val/label', img_path_from='val/A', img_path_to='val/B')))
test_dataloader = dict(
dataset=dict(
type=dataset_type,
data_root=data_root,
data_prefix=dict(seg_map_path='test/label', img_path_from='test/A', img_path_to='test/B')))
train_cfg = dict(type='IterBasedTrainLoop', max_iters=254800, val_interval=1274)
work_dir = r'/new-home/buddhiw/CD-Models/results/changer/levir_cd_test_as_val/work_dir'
|