如何在循环中按类别打印预测

Question

我在打印项目中带有预测标签的图片时遇到问题。

i = 1
for image in DATADIR:
    prediction = model.predict
    ([prepare(r'MY_DIR\manual_testing\{}.jpg'.format(i))])
    img = mpimg.imread(r'MY_DIR\manual_testing\{}.jpg'.format(i))
    imgplot = plt.imshow(img)
    plt.show()
    print(CATEGORIES[int(prediction[0][0])])
    i += 1

此处MY_DIR替换实际目录。我遇到以下错误：TypeError: 'method' object is not subscriptable

我不太明白应该更改什么。如果我尝试将 print('Hello world') 放入 for 循环，一切正常。

p.s。如果您有如何使输出看起来更漂亮的示例，欢迎您。

提前致谢。

Answer 1

我找到了解决方案：

i = 1
for item in os.listdir(DATADIR):
    prediction = model.predict([prepare(r'MY_DIR\manual_testing\{}.jpg'.format(i))])
    img = mpimg.imread(r'MY_DIR\manual_testing\{}.jpg'.format(i))
    imgplot = plt.imshow(img)
    plt.show()

    print(CATEGORIES[int(prediction[0][0])])
    i += 1

如何在循环中按类别打印预测

How to print prediction by category inside a loop

for-loop

python-3.x

conv-neural-network