|
- dataset:
- train_img_root : /dataset/data/polyp/TrainDataset/images/
- train_label_root: /dataset/data/polyp/TrainDataset/masks/
-
- test_CVC-300_img : /dataset/data/polyp/TestDataset/CVC-300/images/
- test_CVC-300_label : /dataset/data/polyp/TestDataset/CVC-300/masks/
-
- test_CVC-ClinicDB_img : /dataset/data/polyp/TestDataset/CVC-ClinicDB/images/
- test_CVC-ClinicDB_label : /dataset/data/polyp/TestDataset/CVC-ClinicDB/masks/
-
- test_CVC-ColonDB_img : /dataset/data/polyp/TestDataset/CVC-ColonDB/images/
- test_CVC-ColonDB_label : /dataset/data/polyp/TestDataset/CVC-ColonDB/masks/
-
- test_ETIS-LaribPolypDB_img : /dataset/data/polyp/TestDataset/ETIS-LaribPolypDB/images/
- test_ETIS-LaribPolypDB_label : /dataset/data/polyp/TestDataset/ETIS-LaribPolypDB/masks/
-
- test_Kvasir_img : /dataset/data/polyp/TestDataset/Kvasir/images/
- test_Kvasir_label : /dataset/data/polyp/TestDataset/Kvasir/masks/
-
- class_num: 18
-
- crop_size :
- w: 352
- h: 352
- batch_size : 32
- num_workers : 8
-
- Train_transform_list:
- resize:
- size: [352, 352]
- random_scale_crop:
- range: [0.75, 1.25]
- random_flip:
- lr: True
- ud: True
- random_rotate:
- range: [0, 359]
- random_image_enhance:
- methods: ['contrast', 'sharpness', 'brightness']
- random_dilation_erosion:
- kernel_range: [2, 5]
- tonumpy: NULL
- normalize:
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- totensor: NULL
-
- Val_transform_list:
- resize:
- size: [352, 352]
- tonumpy: NULL
- normalize:
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- totensor: NULL
-
- #模型
- model:
- model_name : mit_PLD_b2
- is_pretrained : False
- pretrained_path : /dataset/ssformer_L.pth
- from_epoch : 0
-
- training:
- device : cuda
- lr : 1e-4
- max_epoch : 2000
- evl_epoch : 0
-
- other:
- checkpoint_save_path : /model/ssformer_L/
- logger_path : /model/segformer.log
-
- test:
- checkpoint_save_path : /model/ssformer_S.pth
- Test_transform_list:
- resize:
- size: [352, 352]
- tonumpy: NULL
- normalize:
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- totensor: NULL
|