Last active
November 6, 2022 13:23
-
-
Save ritiek/5fa903f97eb6487794077cf3a10f4d3e to your computer and use it in GitHub Desktop.
Keras predicting on all images in a directory
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from keras.models import load_model | |
from keras.preprocessing import image | |
import numpy as np | |
import os | |
# image folder | |
folder_path = '/path/to/folder/' | |
# path to model | |
model_path = '/path/to/saved/model.h5' | |
# dimensions of images | |
img_width, img_height = 320, 240 | |
# load the trained model | |
model = load_model(model_path) | |
model.compile(loss='binary_crossentropy', | |
optimizer='rmsprop', | |
metrics=['accuracy']) | |
# load all images into a list | |
images = [] | |
for img in os.listdir(folder_path): | |
img = os.path.join(folder_path, img) | |
img = image.load_img(img, target_size=(img_width, img_height)) | |
img = image.img_to_array(img) | |
img = np.expand_dims(img, axis=0) | |
images.append(img) | |
# stack up images list to pass for prediction | |
images = np.vstack(images) | |
classes = model.predict_classes(images, batch_size=10) | |
print(classes) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hello, please help. I am getting this error:
cannot identify image file <_io.BytesIO object at 0x7fbe54eb8090>