How can I get the prediction of a single image?

I want just to see the prediction of a single image using the following code:

`

Model reconstruction from JSON file

with open(‘model_struct.json’, ‘r’) as f: model = model_from_json(f.read())

Load weights into the new model

model.load_weights(‘best_weights.h5’)

Load the image file

parser = argparse.ArgumentParser(description=‘Image to test’) parser.add_argument(‘image’, type=str, help=‘The image image file to check’) args = parser.parse_args()

img = load_img(args.image)

Predict

prediction = model.predict(img) `

But it raises the following error: ValueError: Error when checking : expected input_1 to have 4 dimensions, but got array with shape (720, 960, 3)

[eridgd]

You need to have an additional dimension for batch, even with just one image:

img = load_img(args.image)
assert img.ndim == 3
img = np.expand_dims(img, axis=0) # Alternatively could do: img[None, ...]
assert img.ndim == 4

[msohaildanish]

I found the solution for this. I was using utils. load_img() to get image as np.array but in Dronet utils.callback_img() there is extra step before returning image as np array which is return np.asarray(img, dtype=np.float32) * np.float32(1.0/255.0) drone_control/dronet/dronet_perception/src/Dronet/utils.py