|
|
{
|
|
|
"name": "SYSU-test",
|
|
|
"phase": "test",
|
|
|
"gpu_ids":[0],
|
|
|
|
|
|
"path_cd": {
|
|
|
"log": "logs",
|
|
|
"result": "results",
|
|
|
"checkpoint": "checkpoint",
|
|
|
|
|
|
"resume_state": "experiments/sysu_checkpoint/best_cd_model"
|
|
|
},
|
|
|
|
|
|
"datasets": {
|
|
|
"train": {
|
|
|
"name": "SYSU-CD-256",
|
|
|
"datasetroot": "",
|
|
|
"resolution": 256,
|
|
|
"num_workers": 2,
|
|
|
"batch_size": 6,
|
|
|
"use_shuffle": true,
|
|
|
"data_len": -1
|
|
|
},
|
|
|
"val": {
|
|
|
"name": "SYSU-CD-256",
|
|
|
"datasetroot": "",
|
|
|
"resolution": 256,
|
|
|
"num_workers": 2,
|
|
|
"batch_size": 6,
|
|
|
"use_shuffle": true,
|
|
|
"data_len": -1
|
|
|
},
|
|
|
|
|
|
"test": {
|
|
|
"name": "SYSU-CD-256",
|
|
|
"datasetroot": "",
|
|
|
"resolution": 256,
|
|
|
"num_workers": 2,
|
|
|
"batch_size": 6,
|
|
|
"use_shuffle": false,
|
|
|
"data_len": -1
|
|
|
}
|
|
|
},
|
|
|
|
|
|
"model" : {
|
|
|
"name" : "stnr",
|
|
|
"loss" : "ce_dice",
|
|
|
"init_filters" : 16,
|
|
|
"n_classes" : 2,
|
|
|
"mode" : "FUSION",
|
|
|
"conv_mode" : "orignal_dinner",
|
|
|
"local_query_model" : "orignal_dinner",
|
|
|
"up_mode" : "SRCM",
|
|
|
"up_conv_mode" : "deepwise",
|
|
|
"spatial_dims" : 2,
|
|
|
"in_channels" : 3,
|
|
|
"resdiual" : false,
|
|
|
"blocks_down" : [1, 2, 2, 4],
|
|
|
"blocks_up" : [1, 1, 1],
|
|
|
"diff_abs" : "later",
|
|
|
"stage" : 2,
|
|
|
"mamba_act" : "relu",
|
|
|
"norm" : ["GROUP", {"num_groups": 8}]
|
|
|
},
|
|
|
|
|
|
"train" : {
|
|
|
"n_epoch" : 200,
|
|
|
"train_print_iter" : 500,
|
|
|
"val_freq" : 1,
|
|
|
"val_print_iter" : 50,
|
|
|
"optimizer" :{
|
|
|
"type" : "adam",
|
|
|
"lr" : 1e-4
|
|
|
},
|
|
|
"scheduler" : {
|
|
|
"lr_policy" : "linear",
|
|
|
"n_step" : 3,
|
|
|
"gamma" : 0.1
|
|
|
}
|
|
|
}
|
|
|
} |
