ongxuanhong/ssl-in-medical-image-segmentation

Folder Organization

|-data
    |-bccd
        |-images
        |-labels    
    |-lisc
        |-images
        |-labels 
|-processed_data
    |-bccd
        |-images
        |-labels
    |-lisc
        |-images
        |-labels
|-Configs
|-Datasets
|-Models
|-Utils
|-README.md
|-download.ipynb
|-*.py

Download BCCD and LISC data

!gdown --id 1BQdIW_2tfyRbnTahsEYB9Ow4lIekXpSZ

How to run pipeline

python -u fixmatch.py --gpu 0 --seed 1 --exp exp_fixmatch_bbbc --fold 1 --dataset bbbc
--dataset bccd --sup_ratio 0.25 --conf_thres 0.8 --seed 2 --debug 1

How to run grid search

Grid search

{
    "supervised_ratio": [0.01, 0.04],
    "conf_thresh": [0.8, 0.85, 0.9, 0.95],
    "fold": [1, 2, 3, 4, 5],
    "seed": [1, 2, 3]
}

python grid_exp.py