Многократное предсказание за один раз модель CNN

Ниже приведена часть кода прогнозирования с использованием модели CNN, в которой он предсказывает одно прогнозирование за один раз. Может ли кто-нибудь помочь мне, изменив код таким образом, чтобы он предсказывал более одного изображения за раз или предсказал папку, содержащую несколько изображений за раз. Мой код основан на коде, который используется для классификации изображений кошек и собак (2 класса). Мне удалось изменить метку и увеличить количество классов с 2 до 3, но я до сих пор не могу предсказать несколько изображений одновременно.

# Importing image
from IPython.display import Image
import numpy as np
from keras.preprocessing import image
test_image = image.load_img('C:/Users/Hp/data_set/single_prediction/pic11.jpg',target_size = (64, 64)) 
test_image = image.img_to_array(test_image)
test_image = np.expand_dims(test_image, axis = 0)
test_image = np.expand_dims(test_image, axis = 0)
training_set.class_indices
if result[0][0] == 1:
    prediction = 'Mouse'
elif result[0][1] == 1:
    prediction = 'Deer'
elif result[0][2] == 1:
    prediction = 'Lion'
else:
    prediction = 'none'

Image(filename='C:/Users/Hp/data_set/single_prediction/pic11.jpg')

0 ответов

Другие вопросы по тегам