tensorflow
/
hub
mirror of https://github.com/tensorflow/hub.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
hub/examples/image_retraining/README.md

649 B
Raw Permalink Blame History

WARNING: This code is deprecated.

The retrain.py tool from this directory has been replaced by the make_image_classifier tool that gets installed as a command-line tool by the PIP package tensorflow-hub[make_image_classifier]. The new tool uses TensorFlow 2 and supports fine-tuning.

The Colab notebook tf2_image_retraining.ipynb explains the basic technique behind that tool: transfer learning with TF Hub.