ImageAI Laggout
Я обучаю бота ImageAI в Google Colaboratory, используя учебное пособие по направлению к данным. Я использовал код
!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.