Skip to content

Commit c76ed9e

Browse files
committed
update config for nll examples
update config for nll examples
1 parent 3561358 commit c76ed9e

7 files changed

Lines changed: 23 additions & 12 deletions

File tree

seg_nll/JSRT/README.md

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,22 @@ pymic_run train config/unet_gce.cfg
7171
pymic_run test config/unet_gce.cfg
7272
```
7373

74+
### CLSLSR
75+
The CLSLSR method estimates errors in the original noisy label and obtains pixel-level weight maps based on an intial model, and then uses the weight maps to suppress noises in a standard supervised learning procedure. Assume that the initial model is the baseline method, run the following command to obtain the weight maps:
76+
77+
```bash
78+
python clslsr_get_condience config/unet_ce.cfg
79+
```
80+
81+
The weight maps will be saved in `$root_dir/slsr_conf`. Then train the new model and do inference by:
82+
83+
```bash
84+
pymic_run train config/unet_clslsr.cfg
85+
pymic_run test config/unet_clslsr.cfg
86+
```
87+
88+
Note that the weight maps for training images are specified in the configuration file `train_csv = config/data/jsrt_train_mix_clslsr.csv`.
89+
7490
### Co-Teaching
7591
The configuration file for Co-Teaching is `config/unet2d_cot.cfg`. The corresponding setting is:
7692

seg_nll/JSRT/config/unet_ce.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ in_chns = 1
4242
feature_chns = [16, 32, 64, 128, 256]
4343
dropout = [0, 0, 0.3, 0.4, 0.5]
4444
bilinear = False
45-
deep_supervise= False
45+
multiscale_pred = False
4646

4747
[training]
4848
# list of gpus

seg_nll/JSRT/config/unet_clslsr.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ in_chns = 1
4242
feature_chns = [16, 32, 64, 128, 256]
4343
dropout = [0, 0, 0.3, 0.4, 0.5]
4444
bilinear = False
45-
deep_supervise= False
45+
multiscale_pred = False
4646

4747
[training]
4848
# list of gpus

seg_nll/JSRT/config/unet_cot.cfg

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,11 +42,11 @@ in_chns = 1
4242
feature_chns = [16, 32, 64, 128, 256]
4343
dropout = [0, 0, 0.3, 0.4, 0.5]
4444
bilinear = False
45-
deep_supervise= False
45+
multiscale_pred = False
4646

4747
[training]
4848
# list of gpus
49-
gpus = [1]
49+
gpus = [0]
5050

5151
loss_type = CrossEntropyLoss
5252

@@ -63,8 +63,6 @@ ReduceLROnPlateau_patience = 2000
6363

6464
ckpt_save_dir = model/unet_cot
6565

66-
# start iter
67-
iter_start = 0
6866
iter_max = 10000
6967
iter_valid = 100
7068
iter_save = [10000]

seg_nll/JSRT/config/unet_dast.cfg

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,12 +44,11 @@ in_chns = 1
4444
feature_chns = [16, 32, 64, 128, 256]
4545
dropout = [0, 0, 0.3, 0.4, 0.5]
4646
bilinear = False
47-
deep_supervise= False
4847
output_mode = first
4948

5049
[training]
5150
# list of gpus
52-
gpus = [1]
51+
gpus = [0]
5352

5453
loss_type = CrossEntropyLoss
5554

@@ -66,8 +65,6 @@ ReduceLROnPlateau_patience = 2000
6665

6766
ckpt_save_dir = model/unet_dast
6867

69-
# start iter
70-
iter_start = 0
7168
iter_max = 10000
7269
iter_valid = 100
7370
iter_save = [10000]

seg_nll/JSRT/config/unet_gce.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ in_chns = 1
4242
feature_chns = [16, 32, 64, 128, 256]
4343
dropout = [0, 0, 0.3, 0.4, 0.5]
4444
bilinear = False
45-
deep_supervise= False
45+
multiscale_pred = False
4646

4747
[training]
4848
# list of gpus

seg_nll/JSRT/config/unet_trinet.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ in_chns = 1
4242
feature_chns = [16, 32, 64, 128, 256]
4343
dropout = [0, 0, 0.3, 0.4, 0.5]
4444
bilinear = False
45-
deep_supervise= False
45+
multiscale_pred = False
4646

4747
[training]
4848
# list of gpus

0 commit comments

Comments
 (0)