Код логического вывода для прогнозирования в мультиклассовой классификации изображений

Question

Я пытаюсь взять одно входное изображение и предсказать его метку.Изображение обучающих данных было преобразовано в массив и метки в int и преобразовано в один набор данных с использованием DatasetMixin перед подачей в классификатор.Поэтому я преобразовал изображение в массив.

Когда я попробовал с данным кодом ... это ошибка ... Ожидается: in_types [0] .shape [1] == in_types [1] .shape[1] * 1 Факт: 240! = 3

img = cv2.imread('C:/Users/Dell/Desktop/TEST IMAGES/MONOCYTE.jpeg')
plt.imshow(img)
plt.show()

img=np.array((img), dtype = np.float32)
img=img/255.0

x = Variable(np.asarray([img]))
y = model(x)
prediction = y.data.argmax(axis=1)

Answer 1

Детали модели необходимы для точного ответа ...

Но я предполагаю, что модели требуется массив, форма которого (партия, канал, ширина, высота), но формаКажется, что массив, который вы передали в модель (ширина, высота, канал).

Это может быть причиной сообщения об ошибке.