Skip to content

Commit

Permalink
Add link to MobileNet documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
rragundez committed Jun 27, 2018
1 parent dc9ad9b commit b1559b1
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions 05_transfer_learning_for_image_classification.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -144,7 +144,9 @@
"cell_type": "markdown",
"metadata": {},
"source": [
"### Predict image using `MobileNet`"
"### Predict image using [`MobileNet`](https://keras.io/applications/#mobilenet)\n",
"\n",
"The model has been pre-trained on [ImageNet 1000 classes](https://gist.github.com/yrevar/942d3a0ac09ec9e5eb3a)."
]
},
{
Expand Down Expand Up @@ -252,13 +254,23 @@
"save_keras_dataset_to_disk(X_train, y_train, X_test, y_test)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# this only works if you have the tree command\n",
"! tree -L 2 data/CIFAR10"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Use pre-trained `MobileNet` model as base model\n",
"\n",
"load model without last classification layer and set input_shape as closer as possible to original data"
"load model without last classification layer and set input_shape as closer as possible to original data. Not all input shapes work, check the [documentation](https://keras.io/applications/#mobilenet)."
]
},
{
Expand Down

0 comments on commit b1559b1

Please sign in to comment.