| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495 |
- train:
- seed: 42
- epochs: 100
- batch_size: 8
- accum_steps: 1
- amp: true
- num_workers: 8
- device: cuda
- dataset:
- name: lung_ultrasound_seg_multiclass
- root: data/lung_ultrasound
- task_name: multiclass
- image_size: [256, 256]
- in_channels: 3
- num_classes: 4
- class_names: [background, lun, pe, b]
- train_split: train
- val_split: val
- test_split: test
- mask_suffix: .png
- image_suffix: .png
- model:
- name: swin_unet
- encoder_name: swinv2_base_patch4_window12_192_22k
- in_channels: 3
- out_channels: 4
- img_size: 256
- drop_rate: 0.0
- drop_path_rate: 0.2
- pretrain:
- enabled: true
- source: imagenet22k
- checkpoint: weights/swinv2_base_patch4_window12_192_22k.pth
- strict: false
- loss:
- name: generalized_dice_focal
- task_mode: multiclass
- params:
- include_background: false
- metrics:
- task_mode: multiclass
- metrics:
- - name: dice
- - name: miou
- optimizer:
- name: adamw
- lr: 5.0e-5
- weight_decay: 0.05
- betas: [0.9, 0.999]
- scheduler:
- name: cosine
- warmup:
- name: linear
- params:
- start_factor: 0.1
- total_iters: 10
- params:
- T_max: 100
- eta_min: 1.0e-6
- augmentation:
- train:
- random_flip: true
- random_rotate_90: true
- random_resized_crop: false
- random_brightness_contrast: true
- random_gaussian_noise: true
- val:
- center_crop: false
- validation:
- enabled: true
- interval: 1
- metrics: [dice, miou]
- save_best: true
- monitor: dice
- mode: max
- checkpoint:
- dir: outputs/segmentation/train_seg_multiclass
- save_last: true
- save_best_only: false
- keep_top_k: 3
- logging:
- log_interval: 20
- use_tensorboard: true
- tensorboard_dir: outputs/tensorboard/train_seg_multiclass
|
