In my case, the tensorflow version is 1.2, and the corresponding retrain.py is here .
Download and extract color images from here . Now run the retrain.py file as
python retrain.py --image_dir=path\to\dir\where\flowers\images\where\extracted --output_lables=retrained_labels.txt --output_graph=retrained_graph.pb
Note: the last two arguments in the above command are optional.
Now, to test the revised model:
- go to master branch and load the label_image.py code as shown below.

- Then run
python label_image.py --image=image/path/to/test/classfication --graph=retrained_graph.pb --labels=retrained_labels.txt
- The result will look like

source share