IndexError: list index out of range

最新推荐文章于 2024-08-22 22:14:30 发布

da_76

最新推荐文章于 2024-08-22 22:14:30 发布

阅读量207

点赞数

文章标签： python numpy 深度学习

本文链接：https://blog.csdn.net/da_76/article/details/129852003

版权

文章提供了两个Python函数，用于处理Fashion-MNIST数据集的标签和图像显示。`get_fashion_mnist_labels`函数将数字标签转换为对应的文本描述，而`show_images`函数用于展示图像。在示例中，函数被用来显示18个样本图像及其对应的标签，但在运行时遇到了`IndexError:listindexoutofrange`的错误。

摘要由CSDN通过智能技术生成

#两个可视化数据集的函数
def get_fashion_mnist_labels(labels):
"""返回Fashion-MNIST数据集的文本标签"""
text_labels = [ 't-shirt', 'trouser', 'pullover', 'dress', 'coat', 'sandal','shirt'
'sneaker', 'bag', 'ankle boot']
return [text_labels[int(i)] for i in labels]

def show_images(imgs, num_rows, num_cols, titles=None, scale=1.5):
"""plot a list of images."""
figsize = (num_cols * scale, num_rows * scale)
_, axes = d2l.plt.subplots(num_rows, num_cols, figsize=figsize)
axes = axes.flatten()
for i, (ax, img) in enumerate(zip(axes, imgs)):
if torch.is_tensor(img):
#图片张量
ax.imshow(img.numpy())
else:
#PIL图片
ax.imshow(img)
ax.axes.get_xaxis().set_visible(False)
ax.axes.get_yaxis().set_visible(False)
if titles:
ax.set_title(titles[i])
return axes

X, y = next(iter(data.DataLoader(mnist_train, batch_size=18)))
show_images(X.reshape(18, 28, 28), 2, 9, titles=get_fashion_mnist_labels(y));

IndexError                                Traceback (most recent call last)
<ipython-input-31-062a59775dcb> in <module>
      1 #几个样本的图像及其相对应的标签
      2 X, y = next(iter(data.DataLoader(mnist_train, batch_size=18)))
----> 3 show_images(X.reshape(18, 28, 28), 2, 9, titles=get_fashion_mnist_labels(y));
      4 '''X,y = [],[]
      5 #初始化两个列表

<ipython-input-27-9703a7aece05> in get_fashion_mnist_labels(labels)
      4     text_labels = [ 't-shirt', 'trouser', 'pullover', 'dress', 'coat', 'sandal','shirt'
      5         'sneaker', 'bag', 'ankle boot']
----> 6     return [text_labels[int(i)] for i in labels]
      7 
      8 def show_images(imgs, num_rows, num_cols, titles=None, scale=1.5):

<ipython-input-27-9703a7aece05> in <listcomp>(.0)
      4     text_labels = [ 't-shirt', 'trouser', 'pullover', 'dress', 'coat', 'sandal','shirt'
      5         'sneaker', 'bag', 'ankle boot']
----> 6     return [text_labels[int(i)] for i in labels]
      7 
      8 def show_images(imgs, num_rows, num_cols, titles=None, scale=1.5):

IndexError: list index out of range