问题标签 [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 投票
3 回答
3712 浏览

python - Pyinstaller 可执行文件无法导入 torchvision

这是我的main.py

它在命令行中正确运行:python main.py

我需要一个适用于 Windows 的可执行文件。所以我做了 :pyinstaller main.py

但是当我启动main.exe时,里面/dist/main出现了这个错误:

似乎某些源代码没有从 pyinstaller 正确导入。我不确定问题是torch模块还是torchvision

附加信息:

  • 我最近安装了 Visual Studio 2019

系统信息:

  • 窗口 10
  • 蟒蛇 3.7
  • 火炬-1.1.0
  • 火炬视觉-0.3.0

[编辑]

我发现问题出在torchvision中的FrozenBatchNorm2d 类的定义中。以下脚本产生与之前发布的相同的错误:

主文件

我复制了所有的手电筒源文件。但我仍然得到错误...

0 投票
2 回答
1374 浏览

python-2.7 - 在使用 Google Colab 中的所有可用 RAM 后,您的会话崩溃了

我试图让它在 google colab https://github.com/oawiles/X2Face/blob/master/UnwrapMosaic/Face2Face_UnwrapMosaic.ipynb中运行,

我能够让它运行并显示一次结果,但从那以后,我在使用此单元的所有可用 RAM 后,最近我的会话崩溃了

我不确定这是否是 colab 问题

这是日志

0 投票
1 回答
18881 浏览

python-2.7 - 如何在 google colab 中运行 matlab .m 文件

我目前正在尝试运行此 repo https://github.com/Fanziapril/mvfnet ,这需要一个步骤:

是否可以在colab中运行.m文件来执行此操作?

另外,我查看了 oct2py 库 https://blink1073.github.io/oct2py/,但无法成功运行该文件。我遵循了如何在 Python 上运行 MATLAB 代码

0 投票
2 回答
2616 浏览

mnist - MNIST 对象没有属性数据

我正在尝试运行下面给出的代码,但不幸的是我收到以下错误-“MNIST 对象没有属性数据”。错误是因为“mnist_train_set.data.view(-1, 1, 28, 28).float()”这一行。有人可以阐明如何解决这个问题吗?谢谢。

0 投票
2 回答
5417 浏览

python-3.x - 尝试导入 torchvision 时出现导入错误

当我尝试导入 torchvision 时,出现导入错误。

我用谷歌搜索了这个问题,但除了问题可能出在 Cython 之外,什么也没找到。StackOverflow 搜索也没有找到任何东西。我该如何解决这个问题?

Torchvision version is 0.3.0.

Traceback(最近一次调用最后):文件“/home/ai/PycharmProjects/untitled/TRANSFER.py”,第 4 行,在 import torchvision 文件“/home/ai/anaconda3/envs/reinforcement/lib/python3.5/site -packages/torchvision/ init .py”,第 1 行,从 torchvision 导入模型文件“/home/ai/anaconda3/envs/reinforcement/lib/python3.5/site-packages/torchvision/models/init .py ”,第 11 行,从 . 导入检测文件“/home/ai/anaconda3/envs/reinforcement/lib/python3.5/site-packages/torchvision/models/detection/init.py”,第 1 行,从 .faster_rcnn 导入 * 文件“/home/ai/anaconda3/envs/reinforcement/lib/python3.5/site-packages/torchvision/models/detection/faster_rcnn.py”,第 7 行,从 torchvision.ops 导入 misc 作为 misc_nn_ops 文件“/home/ai/anaconda3/envs/reinforcement/lib/python3.5/site-packages/torchvision/ops/init .py ”,第 1 行,从 .boxes 导入 nms , box_iou 文件“/home/ai/anaconda3/envs/reinforcement/lib/python3.5/site-packages/torchvision/ops/boxes.py”,第 2 行,从 torchvision 导入 _C ImportError: /home/ai/anaconda3 /envs/reinforcement/lib/python3.5/site-packages/torchvision/_C.cpython-35m-x86_64-linux-gnu.so:未定义符号:_PyThreadState_UncheckedGet

0 投票
0 回答
410 浏览

python - Pytorch:如何使用参考表创建自定义数据集

我有一个 reference.csv 文件,它包含三列:类型、类和路径。以下是前 5 个示例行:

以更易于查看的格式:

我想创建一个数据集类 (torch.utils.data.Dataset) 来读取图像,这样我就可以使用 DataLoader (torch.utils.data.DataLoader)。

使用参考表创建自定义数据集的正确方法是什么?

0 投票
3 回答
13421 浏览

python - 如何从 DataLoader 获取样本的文件名?

我需要用我训练的卷积神经网络的数据测试结果编写一个文件。数据包括语音数据收集。文件格式需要是“文件名,预测”,但我很难提取文件名。我像这样加载数据:

我正在尝试按如下方式写入文件:

问题os.listdir(TESTH_DATA_PATH + "/all")[i]在于它与加载的文件顺序不同步test_loader。我能做些什么?

0 投票
4 回答
30315 浏览

python - PyTorch - 获取 'TypeError: pic 应该是 PIL Image 或 ndarray。得到' 错误

当我TypeError: pic should be PIL Image or ndarray. Got <class 'numpy.ndarray'>尝试通过. 和DataLoader的版本分别是torch和。Python 的版本在机器上。torchvision1.0.10.2.2.post33.7.1Windows 10

这是代码:

我设置了加载器:

这是完整的错误堆栈跟踪:

0 投票
1 回答
2687 浏览

machine-learning - PyTorch - BCELoss:ValueError:目标和输入必须具有相同数量的元素

当我BCELoss用作我的神经网络的损失函数时,得到ValueError: Target and input must have the same number of elements.

这是我的测试阶段代码(这是一个非常典型的测试阶段代码):

变量的形状output[1000, 10]因为有10目标类(在MNIST数据集中),变量的形状target[1000]因为它包含测试批次的目标类(测试的批量大小设置为10)。那么,问题是如何应用BCELoss作为CNN网络的损失函数?

ps 我使用的数据集是图书馆提供的MNISTtorchvision数据集。

ps这里提供的类似问题的答案并没有为我的案例提出解决方案。

0 投票
1 回答
259 浏览

pytorch - torch.cat 和 torchvision.transforms 的组合导致张量为零

我想在张量的第四层中向图像添加更多信息,前三层基于图像。之后,我想从图像中删除和平(数据增强),并且必须将图像调整为给定大小。

为此,我从一张图片中创建了一个张量,并使用 torch.cat 将它与一个带有一层附加信息的张量连接起来。(几乎但不是所有第二张量的条目都是零。)

我通过 transforms.compose 发送了结果(以剪切和调整张量的大小)但之后张量完全由零组成。

在这里,我建立了一个可重现的示例。

如果我选择“tensor1 = torch.rand(1,height,width)”,我就没有这个问题。但如果大多数条目都是零,我有。使用 scale=(0.5, 1.0) 我也没有问题。

没有一些问题:

  1. 如何使用非零条目调整前三层的大小?

  2. 是我误会了什么,还是真的很奇怪?