Skip to content

Commit b73df02

Browse files
committed
Clarify nucleus sample README
1 parent a9e598f commit b73df02

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

‎samples/nucleus/README.md‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,24 +6,24 @@ The `nucleus.py` file contains the main parts of the code, and the two Jupyter n
66

77

88
## Command line Usage
9-
Train a new model starting from ImageNet weights
9+
Train a new model starting from ImageNet weights using `train` dataset (which is `stage1_train` minus validation set)
1010
```
1111
python3 nucleus.py train --dataset=/path/to/dataset --subset=train --weights=imagenet
1212
```
1313

14-
Train a new model starting from specific weights file
14+
Train a new model starting from specific weights file using the full `stage1_train` dataset
1515
```
16-
python3 nucleus.py train --dataset=/path/to/dataset --subset=train --weights=/path/to/weights.h5
16+
python3 nucleus.py train --dataset=/path/to/dataset --subset=stage1_train --weights=/path/to/weights.h5
1717
```
1818

1919
Resume training a model that you had trained earlier
2020
```
2121
python3 nucleus.py train --dataset=/path/to/dataset --subset=train --weights=last
2222
```
2323

24-
Generate submission file
24+
Generate submission file from `stage1_test` images
2525
```
26-
python3 nucleus.py detect --dataset=/path/to/dataset --subset=train --weights=<last or /path/to/weights.h5>
26+
python3 nucleus.py detect --dataset=/path/to/dataset --subset=stage1_test --weights=<last or /path/to/weights.h5>
2727
```
2828

2929

0 commit comments

Comments
 (0)