ruichaozhu / Transfer-INCEPTIONV3-broccoli

0 stars 0 forks source link

retrain.py is a script that shows how one can adapt a pretrained network for other classification problems. A detailed overview of this script can be found at: https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#0

The script also shows how one can train layers with quantized weights and activations instead of taking a pre-trained floating point model and then quantizing weights and activations. The output graphdef produced by this script is compatible with the TensorFlow Lite Optimizing Converter and can be converted to TFLite format.

We retrain the meta-atom by this code to achieve the classification.