diff --git a/samples/nucleus/README.md b/samples/nucleus/README.md index 5a82f1ebde99ef58b48cb04829e7250ece9d8efa..bdcc521f71e7bd213d74af3fa89c994d089d79d7 100644 --- a/samples/nucleus/README.md +++ b/samples/nucleus/README.md @@ -6,14 +6,14 @@ The `nucleus.py` file contains the main parts of the code, and the two Jupyter n ## Command line Usage -Train a new model starting from ImageNet weights +Train a new model starting from ImageNet weights using `train` dataset (which is `stage1_train` minus validation set) ``` python3 nucleus.py train --dataset=/path/to/dataset --subset=train --weights=imagenet ``` -Train a new model starting from specific weights file +Train a new model starting from specific weights file using the full `stage1_train` dataset ``` -python3 nucleus.py train --dataset=/path/to/dataset --subset=train --weights=/path/to/weights.h5 +python3 nucleus.py train --dataset=/path/to/dataset --subset=stage1_train --weights=/path/to/weights.h5 ``` Resume training a model that you had trained earlier @@ -21,9 +21,9 @@ Resume training a model that you had trained earlier python3 nucleus.py train --dataset=/path/to/dataset --subset=train --weights=last ``` -Generate submission file +Generate submission file from `stage1_test` images ``` -python3 nucleus.py detect --dataset=/path/to/dataset --subset=train --weights= +python3 nucleus.py detect --dataset=/path/to/dataset --subset=stage1_test --weights= ```