Spaces:
Configuration error
Configuration error
File size: 1,060 Bytes
a01ef8c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
# Transfer Learning for Image Classification with TF Hub
This notebook uses transfer learning with multiple [TF Hub](https://tfhub.dev) image classifiers,
[TF datasets](https://www.tensorflow.org/datasets/), and custom image datasets.
The notebook performs the following steps:
1. Import dependencies and setup parameters
1. Prepare the dataset
1. Predict using the original model
1. Transfer Learning
1. Evaluate the model
1. Export the saved model
## Running the notebook
To run the notebook, follow the instructions to setup the [TensorFlow notebook environment](/notebooks/setup.md).
## References
Dataset citations
```
@inproceedings{bossard14,
title = {Food-101 -- Mining Discriminative Components with Random Forests},
author = {Bossard, Lukas and Guillaumin, Matthieu and Van Gool, Luc},
booktitle = {European Conference on Computer Vision},
year = {2014}
}
@ONLINE {tfflowers,
author = "The TensorFlow Team",
title = "Flowers",
month = "jan",
year = "2019",
url = "http://download.tensorflow.org/example_images/flower_photos.tgz" }
```
|