问题标签 [torchvision]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票
1 回答
10170 浏览

python - ImportError:没有名为“torchvision.datasets.mnist”的模块

即使在安装 pytorch 之后,此行也会出现此错误。

0 投票
1 回答
431 浏览

python - 从现有 Torchvision 数据集创建精简数据集

我们都知道包中包含的常见 MNIST 数据集torchvision.datasets。想象一下,我想创建一个仅包含10的数据集的简化版本,以仅对这两个数字进行分类,而不是对所有 10 个值进行分类。

我已经看到可以在继承所需数据集 so 的类中创建自定义数据集__getitem__,它返回给定索引处的项目。所以我这样做了:

问题是我似乎无法返回 None 值,因为它需要“包含张量、数字、字典或列表;找到类‘NoneType’”。

有没有一种简单的方法可以以类似的方式轻松获得此数据集的简化版本?

0 投票
1 回答
254 浏览

python - 在 Ubuntu 18.04 中运行程序(在 VMWare 中运行)时出现此错误:AssertionError: Torch not compiled with CUDA enabled

即使我成功安装了 torchvision 和 cuda,我也收到了这个错误:我在 VMWare(Ubuntu 18.04)中运行程序。

0 投票
1 回答
724 浏览

pytorch - 通过转换将频道添加到 MNIST?

我正在尝试使用 MNIST 数据集。torchvision.datasets它似乎是作为N x H x W (uint8)(批量尺寸、高度、宽度)张量提供的。然而,所有用于处理图像的 pytorch 类(例如Conv2d)都需要一个N x C x H x W (float32)张量,其中C是颜色通道的数量。我尝试添加添加ToTensor变换,但没有添加颜色通道。

有没有办法torchvision.transforms用来添加这个额外的维度?对于 rawtensor我们可以做.unsqueeze(1),但这看起来不是一个非常优雅的解决方案。我只是想以“正确”的方式来做。

这是失败的转换。

0 投票
0 回答
61 浏览

python - Fastai 1.0.x 使用 torchvision 模型

有人可以向我展示一个示例,说明如何在 fastai 中使用 torchvision 模型而不进行修改(即没有进行网络修改create_cnn)?我试过这个,但我得到了RuntimeError: size mismatch。我认为 alexnet 输入大小应该是 224,所以不知道如何调试。

0 投票
2 回答
11402 浏览

python - PyTorch 数据集:将整个数据集转换为 NumPy

我正在尝试将 Torchvision MNIST 训练和测试数据集转换为 NumPy 数组,但找不到实际执行转换的文档。

我的目标是获取整个数据集并将其转换为单个 NumPy 数组,最好不要遍历整个数据集。

我看过How do I turn a Pytorch Dataloader into a numpy array to display image data with matplotlib? 但这并没有解决我的问题。

所以我的问题是,利用torch.utils.data.DataLoader,我将如何将数据集(训练/测试)转换为两个 NumPy 数组,以便所有示例都存在?

注意:我暂时将批量大小保留为默认值 1;我可以将其设置为 60,000 进行训练,将其设置为 10,000 进行测试,但我不想使用这种幻数。

谢谢你。

0 投票
1 回答
1612 浏览

python - Pytorch torchvision MNIST 下载

我是 Pytorch 和 torchvision 的新手。我遵循了一个大约一年前的教程,他试图通过 python 和 torchvision 下载 mnist。

这是如何:

现在我的问题是我收到此错误:

回溯(最近一次通话最后):

下载http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz到 data\MNIST\raw\train-images-idx3-ubyte.gz

文件“C:/Users/Nico/PycharmProjects/PyTorch/mnist.py”,第 13 行,transforms.Normalize((0.1307,), (0.3081,))])),

文件“C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\mnist.py”,第 68 行,在init self.download()

文件“C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\mnist.py”,第 143 行,在下载 download_url(url, root=self.raw_folder, 文件名=文件名,md5=无)

文件“C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\utils.py”,第 73 行,在 download_url reporthook=gen_bar_updater(tqdm())

TypeError:init()缺少1个必需的位置参数:'total'

你们中有人知道我必须改变什么,或者我如何下载/使用它们吗?正如我之前所说,我是新手,我没有任何线索。

我希望你们能帮助我,在此先感谢。

问候尼科又名。万里多

0 投票
1 回答
57 浏览

python - 在 Pytorch Upgrade 之后,我的模型给出了几乎随机的输出

我在“Pytorch 0.4.1”中训练、测试并仍在使用模型。如果我使用音高 0.4.1,它曾经并且仍然可以正常工作(输出应该是)。但是当我升级到 1.0.1 版时,每次我尝试评估相同的输入图像,我都会得到不同的输出(它是回归)。

我试图查看这些版本中发生了什么变化,但由于我没有收到任何错误、警告,所以我不确定我应该具体寻找什么。

PS:我检查了权重,加载模型时它们也相同

0 投票
1 回答
2448 浏览

python - RuntimeError:无效参数 0:张量的大小必须匹配,但维度 0 除外。得到维度 1 中的 3 和 1

在通过 pytorch 训练 resnet50 模型时出现此错误:

RuntimeError:无效参数 0:张量的大小必须匹配,但维度 0 除外。在 /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333 的维度 1 中获得 3 和 1

我正在使用这个:http: //github.com/Helias/Car-Model-Recognition/

使用此数据集http://vmmrdb.cecsresearch.org/

0 投票
1 回答
533 浏览

python-3.x - PyTorch 时尚-MNIST (ETL)