索引器错误:索引0超出轴0的范围,轴大小为0 tensorflow标签检测

2021-02-25 05:05:45 发布

您现在位置:Python中文网/ 问答频道 /正文

嘿,伙计们,我正在使用tensorflow,想传递新的图像,找到它的预测和图像。这个我的代码是:

img = plt.figure(figsize=(32,32), tight_layout={"h_pad":4})
nc=len(set(y_ftest))
for i in range(nc):
    k = np.where(y_ftest==i)
    image = batch_x[k[0][0]]
    prediction = pred.eval(feed_dict={x:X_ftest, y:y_ftest})
    logit = sess.run(tf.argmax(logits, 1), feed_dict={x: X_ftest})
    a = img.add_subplot(nc,2,i+1)
    a.imshow(image, interpolation="None")
    a.set_title(label_dict[logit[0+i]])
    i+=1

这是我得到的错误

^{pr2}$