Как предсказать изображение среди 10 классов вместо двух классов - PullRequest
Новая
       27

Как предсказать изображение среди 10 классов вместо двух классов

0 голосов
/ 20 ноября 2018

Я новичок в области глубокого обучения и пытаюсь предсказать изображение из 10 классов.

Я нашел код ниже, но рассмотрю только два класса.Как я могу настроить код для 10 классов? 

import numpy as np
from keras.preprocessing import image
test_image = image.load_img(path = 'dataset/single_prediction/cat_or_dog_1.jpg', 
                            target_size = (64, 64))
test_image = image.img_to_array(test_image)
test_image = np.expand_dims(test_image, axis = 0)

result = classifier.predict(test_image)
indices = training_set.class_indices
if result[0][0] == 1:
    prediction = 'dog'
else:
    prediction = 'cat'

1 Ответ

0 голосов
/ 20 ноября 2018

Омар, я предлагаю вам немного больше изучить используемые вами библиотеки, такие как Keras (в основном) и Numpy.

Этот тип вопросов будет отвергнут пользователями, поскольку он противоречит лучшим практикам Stack Overflow, поскольку вы просите кого-нибудь дать вам полный код в качестве ответа, не обращая внимания на себя идетализация конкретной задачи, с которой вы сталкиваетесь, с каким классом, а также не детализация всего аспекта вашего кода (например, объекта классификатора, который не детализирован, откуда он взялся, и угадывание не очень хороший способ помочь здесь).

Теперь, переходя к вашей проблеме, я предлагаю:

  1. Лучше понять модель из keras и ее последствия, о том, как загрузить модели: https://keras.io/models/about-keras-models/
  2. Посмотрите внимательнее на документацию keras и ее методы https://keras.io/preprocessing/image/
  3. Посмотрите на этот ответ: https://stackoverflow.com/a/43470074/10642035

Посмотрите на документацию из стекапереполнение, чтобы гарантировать, что вы не потеряете здесь баллы:

...