Skip to content

Commit a123198

Browse files
committed
update configure files and readme, using pymic_net_run
1 parent 184e4e8 commit a123198

7 files changed

Lines changed: 10 additions & 6 deletions

File tree

segmentation/JSRT/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ If you don't want to train the model by yourself, you can download a pre-trained
2222
1. Edit `config/train_test.cfg` by setting the value of `root_dir` as your `JSRT_root`. Then start to train by running:
2323

2424
```bash
25-
pymic_net_run_seg train config/train_test.cfg
25+
pymic_net_run train config/train_test.cfg
2626
```
2727

2828
2. During training or after training, run `tensorboard --logdir model/unet` and you will see a link in the output, such as `http://your-computer:6006`. Open the link in the browser and you can observe the average Dice score and loss during the training stage, such as shown in the following images, where blue and red curves are for training set and validation set respectively. We can observe some over-fitting on the training set.
@@ -35,7 +35,7 @@ pymic_net_run_seg train config/train_test.cfg
3535

3636
```bash
3737
mkdir result
38-
pymic_net_run_seg test config/train_test.cfg
38+
pymic_net_run test config/train_test.cfg
3939
```
4040

4141
2. Then edit `config/evaluation.cfg` by setting `ground_truth_folder_root` as your `JSRT_root`, and run the following command to obtain quantitative evaluation results in terms of dice.

segmentation/JSRT/config/train_test.cfg

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# tensor type (float or double)
33
tensor_type = float
44

5+
task_type = seg
56
root_dir = /home/disk2t/data/JSRT
67
train_csv = config/jsrt_train.csv
78
valid_csv = config/jsrt_valid.csv

segmentation/JSRT2/config/train_test.cfg

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# tensor type (float or double)
33
tensor_type = float
44

5+
task_type = seg
56
root_dir = /home/disk2t/data/JSRT
67
train_csv = ../JSRT/config/jsrt_train.csv
78
valid_csv = ../JSRT/config/jsrt_valid.csv

segmentation/fetal_hc/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ In this example, we use 2D U-Net to segment the fetal brain from ultrasound imag
1717
1. Edit `config/train_test.cfg` by setting the value of `root_dir` as your `HC_root`. Then start to train by running:
1818

1919
```bash
20-
pymic_net_run_seg train config/train_test.cfg
20+
pymic_net_run train config/train_test.cfg
2121
```
2222

2323
2. During training or after training, run `tensorboard --logdir model/unet` and you will see a link in the output, such as `http://your-computer:6006`. Open the link in the browser and you can observe the average Dice score and loss during the training stage, such as shown in the following images, where red and blue curves are for training set and validation set respectively.
@@ -30,7 +30,7 @@ pymic_net_run_seg train config/train_test.cfg
3030

3131
```bash
3232
mkdir result
33-
pymic_net_run_seg test config/train_test.cfg
33+
pymic_net_run test config/train_test.cfg
3434
```
3535

3636
2. Then edit `config/evaluation.cfg` by setting `ground_truth_folder_root` as your `HC_root`, and run the following command to obtain quantitative evaluation results in terms of Dice.

segmentation/fetal_hc/config/train_test.cfg

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# tensor type (float or double)
33
tensor_type = float
44

5+
task_type = seg
56
root_dir = /home/disk2t/data/Fetal_HC
67
train_csv = config/fetal_hc_train.csv
78
valid_csv = config/fetal_hc_valid.csv

segmentation/prostate/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ In this example, we use 3D U-Net to segment the prostate from 3D MRI. First we d
1414
1. Edit `config/train_test.cfg` by setting the value of `root_dir` as your `data/promise12/preprocess`. Then start to train by running:
1515

1616
```bash
17-
pymic_net_run_seg train config/train_test.cfg
17+
pymic_net_run train config/train_test.cfg
1818
```
1919

2020
2. During training or after training, run `tensorboard --logdir model` and you will see a link in the output, such as `http://your-computer:6006`. Open the link in the browser and you can observe the average Dice score and loss during the training stage, such as shown in the following images, where blue and red curves are for training set and validation set respectively.
@@ -27,7 +27,7 @@ pymic_net_run_seg train config/train_test.cfg
2727

2828
```bash
2929
mkdir result
30-
pymic_net_run_seg test config/train_test.cfg
30+
pymic_net_run test config/train_test.cfg
3131
```
3232

3333
2. Then edit `config/evaluation.cfg` by setting `ground_truth_folder_root` as your `data/promise12/preprocess`, and run the following command to obtain quantitative evaluation results in terms of Dice.

segmentation/prostate/config/train_test.cfg

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# tensor type (float or double)
33
tensor_type = float
44

5+
task_type = seg
56
root_dir = /home/disk2t/data/prostate/promise12/preprocess/train
67
train_csv = config/data/image_train.csv
78
valid_csv = config/data/image_valid.csv

0 commit comments

Comments
 (0)