【以下回答由 GPT 生成】
问题分析:
从代码中可以看出,使用了plt.subplot()函数来创建子图,但是最终显示的子图总是不正确。希望得到正确的子图显示结果。
解决方案:
plt.subplot()函数的三个参数分别表示子图布局的行数,列数和当前子图的索引。在代码中,设置的行数为2,列数为4,但是最终显示的子图数量超过了设定的范围。
修改代码的部分如下:
plt.figure(figsize=(16, 8))
for i in range(len(image[:6])):
img = image[:6][i].numpy()
img = np.squeeze(img)
label_img = label[:6][i].numpy()
plt.subplot(2, 3, i+1) # 修改此处的列数为3
plt.title(label_img)
plt.imshow(img)
plt.show() # 将plt.show()放在for循环外,只显示一次图像
如此一来,每次循环时,会自动在指定位置上创建子图,并进行图像显示。一共需要创建2行3列共6个子图来显示图片。
这样,就能够正确显示多个子图的结果了。
请注意,根据图片输出效果,需要根据具体需求调整子图的布局参数。
【相关推荐】
如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^