├── utils └── heheda ├── dataloaders └── heheda ├── networks ├── __init__.py ├── heheda ├── my_visual_tf │ └── __init__.py ├── visual_tf │ └── __init__.py ├── UCMT_networklib │ ├── __init__.py │ └── networkLib │ │ └── backbone │ │ └── __init__.py └── attention_module │ ├── eca_net │ └── __init__.py │ ├── cbam │ └── __init__.py │ ├── triplet_attention │ └── __init__.py │ ├── resnest │ └── model.py │ └── sknet │ └── model.py ├── segmentor ├── __init__.py └── full_supervised │ └── __init__.py ├── README.md └── configs ├── swin_tiny_patch4_window7_224_lite.yaml ├── acdc ├── percent_20 │ ├── em_config.yaml │ ├── unet_config.yaml │ ├── DCT_2D_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── scp_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ ├── mt_config.yaml │ ├── ssnet_config.yaml │ └── uamt_config.yaml ├── percent_30 │ ├── em_config.yaml │ ├── unet_config.yaml │ ├── DCT_2D_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── scp_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ ├── mt_config.yaml │ ├── ssnet_config.yaml │ └── uamt_config.yaml ├── percent_40 │ ├── em_config.yaml │ ├── unet_config.yaml │ ├── DCT_2D_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── scp_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ ├── mt_config.yaml │ ├── ssnet_config.yaml │ └── uamt_config.yaml ├── percent_5 │ ├── em_config.yaml │ ├── unet_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── scp_config.yaml │ ├── DCT_2D_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ ├── mt_config.yaml │ ├── uamt_config.yaml │ └── ssnet_config.yaml ├── percent_50 │ ├── em_config.yaml │ ├── unet_config.yaml │ ├── DCT_2D_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── scp_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ ├── mt_config.yaml │ ├── ssnet_config.yaml │ └── uamt_config.yaml ├── percent_60 │ ├── em_config.yaml │ ├── unet_config.yaml │ ├── DCT_2D_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── scp_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ ├── mt_config.yaml │ ├── ssnet_config.yaml │ └── uamt_config.yaml └── percent_10 │ └── em_config.yaml ├── BML_noaug ├── percent_10 │ ├── unet_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── em_config.yaml │ ├── DCT_2D_config.yaml │ ├── scp_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ ├── mt_config.yaml │ └── uamt_config.yaml ├── percent_20 │ ├── unet_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── em_config.yaml │ ├── DCT_2D_config.yaml │ ├── scp_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ ├── mt_config.yaml │ └── uamt_config.yaml ├── percent_30 │ ├── unet_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── em_config.yaml │ ├── DCT_2D_config.yaml │ ├── scp_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ └── mt_config.yaml ├── percent_40 │ ├── unet_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── em_config.yaml │ ├── DCT_2D_config.yaml │ ├── scp_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ └── mt_config.yaml ├── percent_5 │ ├── unet_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── em_config.yaml │ ├── DCT_2D_config.yaml │ ├── scp_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ └── mt_config.yaml ├── percent_50 │ ├── unet_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── em_config.yaml │ ├── DCT_2D_config.yaml │ ├── scp_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ └── mt_config.yaml └── percent_60 │ ├── unet_config.yaml │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── em_config.yaml │ ├── DCT_2D_config.yaml │ ├── scp_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ ├── ssm4mis_config.yaml │ └── mt_config.yaml ├── BUSI_original ├── percent_10 │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── unet_config.yaml │ ├── em_config.yaml │ ├── scp_config.yaml │ ├── DCT_2D_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ └── ssm4mis_config.yaml ├── percent_20 │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── unet_config.yaml │ ├── em_config.yaml │ ├── scp_config.yaml │ ├── DCT_2D_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ └── ssm4mis_config.yaml ├── percent_30 │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── unet_config.yaml │ ├── em_config.yaml │ ├── scp_config.yaml │ ├── DCT_2D_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ └── ssm4mis_config.yaml ├── percent_40 │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── unet_config.yaml │ ├── em_config.yaml │ ├── scp_config.yaml │ ├── DCT_2D_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ └── ssm4mis_config.yaml ├── percent_5 │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── unet_config.yaml │ ├── em_config.yaml │ ├── scp_config.yaml │ ├── DCT_2D_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ └── ssm4mis_config.yaml ├── percent_50 │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── unet_config.yaml │ ├── em_config.yaml │ ├── scp_config.yaml │ ├── DCT_2D_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ └── ssm4mis_config.yaml └── percent_60 │ ├── cct_config.yaml │ ├── cps_config.yaml │ ├── unet_config.yaml │ ├── em_config.yaml │ ├── scp_config.yaml │ ├── DCT_2D_config.yaml │ ├── ssnet_config.yaml │ ├── urpc_config.yaml │ └── ssm4mis_config.yaml └── BML_augmerge_1674 ├── percent_10 ├── cct_config.yaml ├── cps_config.yaml └── unet_config.yaml ├── percent_20 ├── cct_config.yaml ├── cps_config.yaml └── unet_config.yaml ├── percent_30 ├── cct_config.yaml ├── cps_config.yaml └── unet_config.yaml ├── percent_40 ├── cct_config.yaml ├── cps_config.yaml └── unet_config.yaml ├── percent_5 ├── cct_config.yaml ├── cps_config.yaml └── unet_config.yaml ├── percent_50 ├── cct_config.yaml ├── cps_config.yaml └── unet_config.yaml ├── percent_60 ├── cct_config.yaml ├── cps_config.yaml └── unet_config.yaml └── full_supervised └── unet_config.yaml /utils/heheda: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /dataloaders/heheda: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/heheda: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /segmentor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/my_visual_tf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/visual_tf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/UCMT_networklib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentor/full_supervised/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/UCMT_networklib/networkLib/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | from . import resnet -------------------------------------------------------------------------------- /networks/attention_module/eca_net/__init__.py: -------------------------------------------------------------------------------- 1 | """ 2 | refer to https://github.com/BangguWu/ECANet 3 | """ -------------------------------------------------------------------------------- /networks/attention_module/cbam/__init__.py: -------------------------------------------------------------------------------- 1 | """ 2 | refer to https://github.com/luuuyi/CBAM.PyTorch 3 | """ 4 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # Official code for Prototype Contrastive Consistency Learning for Semi-Supervised Medical Image Segmentation 2 | -------------------------------------------------------------------------------- /networks/attention_module/triplet_attention/__init__.py: -------------------------------------------------------------------------------- 1 | """ 2 | refer to https://github.com/LandskapeAI/triplet-attention 3 | """ -------------------------------------------------------------------------------- /networks/attention_module/resnest/model.py: -------------------------------------------------------------------------------- 1 | import timm 2 | import torch.nn as nn 3 | 4 | 5 | def resnest50(n_classes=1000): 6 | model = timm.create_model('resnest50d', pretrained=False) 7 | model.fc = nn.Linear(2048, n_classes) 8 | return model 9 | -------------------------------------------------------------------------------- /networks/attention_module/sknet/model.py: -------------------------------------------------------------------------------- 1 | import timm 2 | import torch.nn as nn 3 | 4 | 5 | def sknet50(n_classes=1000): 6 | model = timm.create_model('skresnet50', pretrained=False) 7 | model.fc = nn.Linear(2048, n_classes) 8 | return model 9 | 10 | 11 | if __name__ == '__main__': 12 | import torch 13 | net = sknet50() 14 | net(torch.zeros(32, 3, 224, 224)) -------------------------------------------------------------------------------- /configs/swin_tiny_patch4_window7_224_lite.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: swin 3 | NAME: swin_tiny_patch4_window7_224 4 | DROP_PATH_RATE: 0.2 5 | PRETRAIN_CKPT: "../code/pretrained_ckpt/swin_tiny_patch4_window7_224.pth" 6 | SWIN: 7 | FINAL_UPSAMPLE: "expand_first" 8 | EMBED_DIM: 96 9 | DEPTHS: [ 2, 2, 2, 2 ] 10 | DECODER_DEPTHS: [ 2, 2, 2, 1] 11 | NUM_HEADS: [ 3, 6, 12, 24 ] 12 | WINDOW_SIZE: 7 13 | PATCH_SIZE: 4 #按照224 / 4==56 往下除以2 -------------------------------------------------------------------------------- /configs/acdc/percent_20/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 28.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet_em 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 28.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 42.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet_em 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 42.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 56.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet_em 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 56.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 7.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet_em 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 7.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 7.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet_em 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 7.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 84.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet_em 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: 6 | - 0 7 | - 1 8 | - 2 9 | - 3 10 | root_dir: /data/heshihuan/datasets/ACDC 11 | in_channels: 1 12 | kwargs: 13 | labeled_num: 84.0 14 | labeled_bs: 8 15 | batch_size: 16 16 | batch_size_val: 1 17 | num_workers: 16 18 | resize_size: 19 | - 256 20 | - 256 21 | network: 22 | net_type: unet 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.01 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cct 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cps 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_scp 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: DCT_2D 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cct 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cps 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_scp 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_urpc 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: DCT_2D 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cct 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cps 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_scp 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_urpc 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: DCT_2D 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cct 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cps 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_scp 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_urpc 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: DCT_2D 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_urpc 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: DCT_2D 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cct 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cps 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_scp 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_urpc 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: DCT_2D 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cct 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_cps 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_scp 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_urpc 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 224 21 | - 224 22 | network: 23 | net_type: ssm4mis_cnn_tf 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 224 21 | - 224 22 | network: 23 | net_type: ssm4mis_cnn_tf 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 224 21 | - 224 22 | network: 23 | net_type: ssm4mis_cnn_tf 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 224 21 | - 224 22 | network: 23 | net_type: ssm4mis_cnn_tf 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 224 21 | - 224 22 | network: 23 | net_type: ssm4mis_cnn_tf 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 224 21 | - 224 22 | network: 23 | net_type: ssm4mis_cnn_tf 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.1 28 | consistency_rampup: 1000.0 29 | save: 30 | val_interval: 100 31 | max_iterations: 20000 32 | cur_program_log_root: null 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: null 40 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_em 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/scp_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_scp 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_mt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_mt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_mt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_mt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/uamt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_uamt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_mt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_mt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/DCT_2D_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: DCT_2D 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_ssnet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | magnitude: 6.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/urpc_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_urpc 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/acdc/percent_10/em_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | dimension: 3D 4 | num_classes: 4 5 | label_list: [0,1,2,3] 6 | # root_dir: /home/lu/HeSH/datasets/ACDC/ACDC 7 | root_dir: /data/heshihuan/datasets/ACDC 8 | 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 14 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: [256 , 256] 17 | 18 | network: 19 | net_type: unet_em 20 | lr: 21 | base_lr: 0.05 22 | 23 | 24 | loss: 25 | consistency: 0.01 26 | consistency_rampup: 1000.0 27 | 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | 33 | logging: 34 | logfile_level: info 35 | stdout_level: info 36 | log_file: info_ 37 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 38 | rewrite: true 39 | log_to_file: 40 | 41 | 42 | 43 | 44 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_ssnet 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | magnitude: 6.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_20/uamt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 28.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_uamt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_ssnet 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | magnitude: 6.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_30/uamt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 42.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_uamt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_ssnet 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | magnitude: 6.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_40/uamt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 56.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_uamt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_5/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_ssnet 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | magnitude: 6.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_ssnet 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | magnitude: 6.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_50/uamt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 7.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_uamt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/ssnet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_ssnet 24 | lr: 25 | base_lr: 0.05 26 | loss: 27 | consistency: 0.01 28 | consistency_rampup: 1000.0 29 | magnitude: 6.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/acdc/percent_60/uamt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: ACDC_semi 3 | subset_name: null 4 | dimension: 3D 5 | num_classes: 4 6 | label_list: 7 | - 0 8 | - 1 9 | - 2 10 | - 3 11 | root_dir: /data/heshihuan/datasets/ACDC 12 | in_channels: 1 13 | kwargs: 14 | labeled_num: 84.0 15 | labeled_bs: 8 16 | batch_size: 16 17 | batch_size_val: 1 18 | num_workers: 16 19 | resize_size: 20 | - 256 21 | - 256 22 | network: 23 | net_type: unet_uamt 24 | ema_alpha: 0.99 25 | lr: 26 | base_lr: 0.05 27 | loss: 28 | consistency: 0.01 29 | consistency_rampup: 1000.0 30 | save: 31 | val_interval: 100 32 | max_iterations: 20000 33 | cur_program_log_root: null 34 | logging: 35 | logfile_level: info 36 | stdout_level: info 37 | log_file: info_ 38 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 39 | rewrite: true 40 | log_to_file: null 41 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_10/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_10/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_10/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.1 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_20/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_20/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_20/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.2 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_30/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_30/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_30/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.3 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_40/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_40/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_40/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.4 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_5/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_5/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_5/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.05 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_50/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_50/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_50/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.5 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_60/cct_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cct 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_60/cps_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet_cps 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/percent_60/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_semi 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 0.6 12 | labeled_bs: 8 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 256 18 | - 256 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_10/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_20/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_30/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_40/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_5/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_50/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BUSI_original/percent_60/ssm4mis_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: BUSI_original_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BUSI/train_val_test_256X256_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 224 19 | - 224 20 | network: 21 | net_type: ssm4mis_cnn_tf 22 | lr: 23 | base_lr: 0.05 24 | loss: 25 | consistency: 0.1 26 | consistency_rampup: 1000.0 27 | save: 28 | val_interval: 100 29 | max_iterations: 20000 30 | cur_program_log_root: null 31 | logging: 32 | logfile_level: info 33 | stdout_level: info 34 | log_file: info_ 35 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 36 | rewrite: true 37 | log_to_file: null 38 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_mt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_mt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_30/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.3 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_mt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_40/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.4 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_mt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_5/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.05 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_mt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_50/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.5 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_mt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_60/mt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.6 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_mt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_augmerge_1674/full_supervised/unet_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_MRI_augmerge_1674_full_supervised 3 | dimension: 2D 4 | num_classes: 2 5 | label_list: 6 | - 0 7 | - 1 8 | root_dir: /data/heshihuan/datasets/BML_cropresize_1674/merge_modality_tr_val_test_7_1_2 9 | in_channels: 1 10 | kwargs: 11 | labeled_num: 1.0 12 | labeled_bs: 16 13 | batch_size: 16 14 | batch_size_val: 1 15 | num_workers: 16 16 | resize_size: 17 | - 224 18 | - 224 19 | network: 20 | net_type: unet 21 | lr: 22 | base_lr: 0.05 23 | loss: 24 | consistency: 0.1 25 | consistency_rampup: 1000.0 26 | save: 27 | val_interval: 100 28 | max_iterations: 20000 29 | cur_program_log_root: null 30 | logging: 31 | logfile_level: info 32 | stdout_level: info 33 | log_file: info_ 34 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 35 | rewrite: true 36 | log_to_file: null 37 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_10/uamt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.1 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_uamt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | -------------------------------------------------------------------------------- /configs/BML_noaug/percent_20/uamt_config.yaml: -------------------------------------------------------------------------------- 1 | dataset: 2 | exp: breast_BML_noaug_semi 3 | subset_name: null 4 | dimension: 2D 5 | num_classes: 2 6 | label_list: 7 | - 0 8 | - 1 9 | root_dir: /data/heshihuan/datasets/BML/merge_train_val_test_7_1_2 10 | in_channels: 1 11 | kwargs: 12 | labeled_num: 0.2 13 | labeled_bs: 8 14 | batch_size: 16 15 | batch_size_val: 1 16 | num_workers: 16 17 | resize_size: 18 | - 256 19 | - 256 20 | network: 21 | net_type: unet_uamt 22 | ema_alpha: 0.99 23 | lr: 24 | base_lr: 0.05 25 | loss: 26 | consistency: 0.1 27 | consistency_rampup: 1000.0 28 | save: 29 | val_interval: 100 30 | max_iterations: 20000 31 | cur_program_log_root: null 32 | logging: 33 | logfile_level: info 34 | stdout_level: info 35 | log_file: info_ 36 | log_format: '%(asctime)s %(levelname)-7s %(message)s' 37 | rewrite: true 38 | log_to_file: null 39 | --------------------------------------------------------------------------------