Отображение кода и маркировка кода Отображает только: - PullRequest
0 голосов
/ 07 октября 2019

Я делаю классификатор изображений с использованием TensorFlow на основе учебника по https://blog.francium.tech/build-your-own-image-classifier-with-tensorflow-and-keras-dc147a15e38e. Я обучил свой классификатор, и сейчас я работаю над кодом для маркировки набора тестируемых изображений. Вот код:

fig = plt.figure(figsize=(14,14))

for cnt, data in enumerate(testing_images[10:40]):

    y = fig.add_subplot(6,5, cnt+1)
    img = data[0]
    data = img.reshape(1,64,64,1)
    model_out = model.predict([data])

    if np.argmax(model_out == 1):
        str_label = "Airplane"
    else:
        str_label = "Rocket"

    y.imshow(img, cmap="gray")
    plt.title(str_label)
    y.axes.get_xaxis().set_visible(False)
    y.axes.get_yaxis().set_visible(False)

Вместо того, чтобы показывать каждое изображение в массиве testing_images, это единственная напечатанная вещь: <Figure size 1008x1008 with 0 Axes> Кто-нибудь знает, почему это может быть?

...