Code from the keras samples, with TensorFlow imports added. https://keras.io
In [ ]:
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import tensorflow as tf
import numpy as np
from tensorflow.contrib.keras.python.keras.preprocessing import image
from tensorflow.contrib.keras.python.keras.applications.resnet50 import *
In [ ]:
print (tf.__version__) # Must be v1.1+
In [ ]:
model = ResNet50(weights='imagenet')
img = image.load_img('deer.jpg', target_size=(224, 224))
x = image.img_to_array(img)
x = np.expand_dims(x, axis=0)
x = preprocess_input(x)
preds = model.predict(x)
print('Predicted:', decode_predictions(preds, top=5)[0])