New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

batch size when predicting #572

Open

vuhungtvt2018 opened this issue Jul 1, 2023 · 1 comment

vuhungtvt2018 commented Jul 1, 2023

how can I apply batch size when predicting multible image
plsease help me!
thanks!

tshr-d-dragon commented Sep 21, 2023

@vuhungtvt142 you can try this code snippet:

from glob import glob
import numpy as np
import cv2

images_paths = glob("/*")
images_array = []

for image_path in images_paths:
    image = cv2.imread(image_path)[:,:,::-1]
    images.append(image)

images_array = np.asarray(images_array).reshape(len(images_paths), H, W, N_CHANNELS)

predictions = model.predict(images_array)

Please let me know if this works!
Also, close the issue if this solves the problem!!!:)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment