File size: 611 Bytes
e4f1a53 6e6ecde e4f1a53 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | import helper.optimizer_def
from keras.models import load_model
import numpy as np
CONVERT_CLASS_PRED_TO_NAME = ["Common Rust", "Gray Leaf Spot", "Leaf Blight"]
def fetch_model(opt_name: str):
return load_model(f'models/{opt_name}-model-001.h5', safe_mode=False)
def preprocess_image(img):
img = img.resize((224, 224))
img_array = np.array(img)
img_array = np.expand_dims(img_array, axis=0)
return img_array
def classify_image(model, img):
prediction = model.predict(img)
predicted_class = np.argmax(prediction, axis=1)[0]
return CONVERT_CLASS_PRED_TO_NAME[predicted_class]
|