Arkajyoti Mukherjee
Oct 2 · 1 min read

MasayoMusic

you can change the input shape as described in the article while creating the keras.Sequential layer and adding the hub model.

classifier_url =”https://tfhub.dev/google/tf2-preview/mobilenet_v2/classification/2"IMAGE_SHAPE = (224, 224) #choose your own input shapeclassifier = tf.keras.Sequential([ 
hub.KerasLayer(classifier_url, input_shape=IMAGE_SHAPE+(3,))
])
    Arkajyoti Mukherjee

    Written by

    Crazy coder with an even more crazy writer inside :3

    Welcome to a place where words matter. On Medium, smart voices and original ideas take center stage - with no ads in sight. Watch
    Follow all the topics you care about, and we’ll deliver the best stories for you to your homepage and inbox. Explore
    Get unlimited access to the best stories on Medium — and support writers while you’re at it. Just $5/month. Upgrade