# Path to save the model h5 file. model_fname = os.path.join(model_path, "model.h5") os.makedirs(model_path, exist_ok=True) img_height = 299 model = Net(weights="imagenet", input_shape=(img_height, img_height, 3)) # Load the image for prediction. img = image.load_img(img_path, target_size=(img_height, img_height)) x = image.img_to_array(img) x = np.expand_dims(x, axis=0) x = preprocess_input(x) preds = model.predict(x) # decode the results into a list of tuples (class, description, probability) # (one such list for each sample in the batch) print("Predicted:", decode_predictions(preds, top=3)[0]) # Predicted: [(u'n02504013', u'Indian_elephant', 0.82658225), (u'n01871265', u'tusker', 0.1122357), (u'n02504458', u'African_elephant', 0.061040461)] # Save the h5 file to path specified. model.save(model_fname) # ## Benchmark Keras prediction speed. # In[2]: import time times = []