时间戳

首页 > 解决方案 > 如何在循环内按类别打印预测

python-3.x - 如何在循环内按类别打印预测

问题描述

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

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 循环，一切正常。

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

先感谢您。

标签： python-3.xfor-loopconv-neural-network

解决方案

我找到了解决方案：

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

推荐阅读