ImageAI Laggout - PullRequest
Новая
       28

ImageAI Laggout

0 голосов
/ 16 января 2020

Я тренирую бота ImageAI в Google Colab Laboratory, используя учебник intodatascience . Я использовал код 

!pip3 install tensorflow
!pip3 install opencv-python
!pip3 install keras
!pip3 install imageai --upgrade

, чтобы установить пакеты, 

from google.colab import drive
drive.mount('/content/drive')

, чтобы смонтировать диск, и 

from imageai.Prediction.Custom import ModelTraining
model_trainer = ModelTraining()
model_trainer.setModelTypeAsResNet()
model_trainer.setDataDirectory("/<MY DRIVE PATH>/F360")
model_trainer.trainModel(num_objects=120, num_experiments=200, enhance_data=False, batch_size=32, show_network_summary=True)

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

ValueError: A target array with shape (32, 123) was passed for an output of shape (None, 120) while using as loss `categorical_crossentropy`. This loss expects targets to have the same shape as the output.
...