MobileNET_hub.py

import tensorflow as tf
import tensorflow_hub as hub 

import numpy as np
from skimage import data

module = hub.Module("https://tfhub.dev/google/imagenet/mobilenet_v2_140_224/classification/2")
height, width = hub.get_expected_image_size(module)

cat = data.chelsea()
images = np.array([cat])

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    images = tf.cast(images, tf.float32)
    images = images/255
    images = tf.image.resize_images(images, (height, width))
    classes = sess.run(module(images))

np.argmax(classes)