Как предсказать некоторые изображения по обученной модели на Tensorflow - PullRequest
0 голосов
/ 01 марта 2020

Я супер новичок. Я проверил официальный сайт Передача обучения с предварительно обученным Con vNet. Я хотел бы делать прогнозы по обученной модели сайта. Следующий код неверен? И я хотел бы знать это «изображение» и «имя класса». Хм .. Как мне быть? ... Пожалуйста, дайте мне совет ...

predictions = model.predict(test_batches)[1]
print(predictions)
#[-0.11642772]

1 Ответ

1 голос
/ 02 марта 2020

Прежде всего, Tensorflow ужасен. Невероятно подвержен ошибкам и сложен в установке и использовании. Я бы рекомендовал вместо этого использовать учебники по Pytorch. Во-вторых, у размещенной вами ссылки есть кнопка Colab. Вы пытались нажать на эту кнопку? Colab облегчает установку, потому что он подключен к Интернету, а не на вашем компьютере.

Кроме того, обучение передаче НЕ является новичком topi c. Возможно, попробуйте более простой блокнот, если он вам подходит: https://github.com/sgrvinod/Deep-Tutorials-for-PyTorch/blob/master/README.md

...