eligapris/maize-diseases-detection

1 2# TF for image classification model3 4import tensorflow5import numpy6from PIL import Image7 8model = tensorflow.saved_model.load('./')9classes = [ "Healthy" ,  "Gray_Leaf_Spot" ,  "Blight" ,  "Common_Rust" , ]10 11img = Image.open("image.jpg").convert('RGB')12img = img.resize((300, 300 * img.size[1] // img.size[0]), Image.ANTIALIAS)13inp_numpy = numpy.array(img)[None]14 15 16inp = tensorflow.constant(inp_numpy, dtype='float32')17 18class_scores = model(inp)[0].numpy()19 20 21print("")22print("class_scores", class_scores)23print("Class : ", classes[class_scores.argmax()])