Skip to content

Latest commit

 

History

History

image_retraining

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

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.