20 lines
550 B
Python
20 lines
550 B
Python
import numpy as np
|
|
from keras.utils import load_img, img_to_array
|
|
from keras.applications.vgg16 import preprocess_input, decode_predictions
|
|
|
|
def predict(model, path):
|
|
# receive image path as CLI argument
|
|
img = load_img(path ,color_mode='rgb', target_size=(224, 224))
|
|
|
|
# loaded image to np array for model to read
|
|
x = img_to_array(img)
|
|
x.shape
|
|
x = np.expand_dims(x, axis=0)
|
|
|
|
# process array and make predictions
|
|
x = preprocess_input(x)
|
|
features = model.predict(x)
|
|
p = decode_predictions(features)
|
|
|
|
return p
|