m0_55193846 2021-08-01 16:54
浏览 274
已结题

torch 载入mnist数据集进行图片拼接显示出现

报错如下
Traceback (most recent call last):
File "C:\Users\杨颖\Downloads\12805831_IPPkFxJhRMBDIRWZxeeecJnzZ.py", line 79, in
imshow(torchvision.utils.make_grid(iter(train_loader).next()[0]))
File "D:\python\lib\site-packages\torch\utils\data\dataloader.py", line 521, in next
data = self._next_data()
File "D:\python\lib\site-packages\torch\utils\data\dataloader.py", line 561, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "D:\python\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\python\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\python\lib\site-packages\torchvision\datasets\mnist.py", line 126, in getitem
img, target = self.data[index], int(self.targets[index])
IndexError: index 25934 is out of bounds for dimension 0 with size 10000
**代码如下

```python
import torch
import torch.nn as nn
import torch.optim as optim
import torchvision
import os
#load data
transform = torchvision.transforms.ToTensor() #定义数据预处理方式:转换 PIL.Image 成 torch.FloatTensor

train_data = torchvision.datasets.MNIST(root="F:\python_mnist", #数据目录,这里目录结构要注意。
train=True, #是否为训练集
transform=transform, #加载数据预处理
download=False) #是否下载
test_data = torchvision.datasets.MNIST(root="F:\python_mnist",
train=False,
transform=transform,
download=False)
#数据加载器:组合数据集和采样器;batch_size=64:同时并行处理64张图片
train_loader = torch.utils.data.DataLoader(dataset = train_data,batch_size = 64,shuffle = True)
test_loader = torch.utils.data.DataLoader(dataset = test_data,batch_size = 64,shuffle = False)

#展示数据/图像
import numpy as np
import matplotlib.pyplot as plt
def imshow(img):
img = img / 2 + 0.5 # unnormalize
npimg = img.numpy()
plt.imshow(np.transpose(npimg, (1, 2, 0)))
plt.show()

torchvision.utils.make_grid 将图片进行拼接

imshow(torchvision.utils.make_grid(iter(train_loader).next()[0]))

```:**

在最后一句报错,,是不是应该打开mnist.py 把index的范围扩大啊,可是应该怎么改

  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 8月9日
    • 创建了问题 8月1日

    悬赏问题

    • ¥30 STM32 INMP441无法读取数据
    • ¥100 求汇川机器人IRCB300控制器和示教器同版本升级固件文件升级包
    • ¥15 用visualstudio2022创建vue项目后无法启动
    • ¥15 x趋于0时tanx-sinx极限可以拆开算吗
    • ¥500 把面具戴到人脸上,请大家贡献智慧
    • ¥15 任意一个散点图自己下载其js脚本文件并做成独立的案例页面,不要作在线的,要离线状态。
    • ¥15 各位 帮我看看如何写代码,打出来的图形要和如下图呈现的一样,急
    • ¥30 c#打开word开启修订并实时显示批注
    • ¥15 如何解决ldsc的这条报错/index error
    • ¥15 VS2022+WDK驱动开发环境