Вопросы по теме 'torchvision'

несоответствие размеров, m1: [3584 x 28], m2: [784 x 128] в /pytorch/aten/src/TH/generic/THTensorMath.cpp:940
Я выполнил следующий код и получил ошибку, показанную в самом низу. Я хотел бы знать, как решить эту проблему. Благодарность import torch.nn as nn import torch.nn.functional as F from torch import optim from torchvision import transforms _tasks...
8371 просмотров
schedule 06.12.2022

Добавить канал в MNIST через преобразование?
Я пытаюсь использовать набор данных MNIST из torchvision.datasets . Кажется, он предоставляется как тензор N x H x W (uint8) (размер партии, высота, ширина). Однако для всех классов pytorch для работы с изображениями (например, Conv2d )...
584 просмотров
schedule 25.02.2022

Наборы данных PyTorch: преобразование всего набора данных в NumPy
Я пытаюсь преобразовать наборы данных для обучения и тестирования Torchvision MNIST в массивы NumPy, но не могу найти документацию для фактического выполнения преобразования. Моя цель - взять весь набор данных и преобразовать его в один массив...
8440 просмотров
schedule 10.05.2023

PyTorch - BCELoss: ValueError: цель и вход должны иметь одинаковое количество элементов
Когда я использую BCELoss в качестве функции потерь моей нейронной сети, получаю ValueError: Target and input must have the same number of elements . Вот мой код для фазы тестирования (это довольно типичный код фазы тестирования):...
2259 просмотров

Модуль Pytorch не найден
Я использую Python 3.6 и пытаюсь использовать Pytorch. Я удалил его с помощью pip3 и переустановил Hugos-MacBook-Pro-2:project hugokitano$ pip3 install torch Requirement already satisfied: torch in /usr/local/lib/python3.7/site-packages (1.3.1)...
1333 просмотров
schedule 18.06.2023

Как я могу превратить тензор факела в список массивов numpy в Pytorch?
У меня есть тензор факела, который выглядит как torch.Size ([32, 3, 64, 64]). Я пытаюсь преобразовать тензор во что-то, что может передавать эти утверждения: assert(type(images) == list) assert(type(images[0]) == np.ndarray)...
2304 просмотров
schedule 25.02.2022

PyTorch DataLoader добавляет дополнительное измерение для TorchVision MNIST
Я новичок в PyTorch и экспериментировал с классом DataLoader. Когда я пытаюсь загрузить набор данных MNIST, DataLoader добавляет дополнительное измерение после пакетного измерения. Я не уверен, что вызывает это. import torch from...
851 просмотров
schedule 15.03.2023

Простой способ оценить Torchvision Pytorch на одном изображении
У меня есть предварительно обученная модель на Pytorch v1.3, torchvision v0.4.2 следующим образом: import PIL, torch, torchvision # Load and normalize the image img_file = "./robot_image.jpg" img = PIL.Image.open(img_file) img =...
795 просмотров
schedule 20.05.2022

Как разделить данные на обучающие и тестовые наборы из одного каталога с помощью PyTorch?
У меня есть папка с данными, в которой данные не разделены на папки для обучения и тестирования. Как разделить данные на наборы для обучения и тестирования? Метки берутся из имен файлов, поэтому любое изменение в этом порядке должно включать метки....
789 просмотров

RuntimeError: ожидается 4-мерный ввод для 4-мерного веса [64, 3, 7, 7], но вместо этого получен 3-мерный ввод размера [3, 32, 32].
Я новичок в PyTorch и нейронных сетях в целом. Я пытался реализовать модель resnet-50 от torchvision в наборе данных CIFAR-10. import torchvision import torch import torch.nn as nn from torch import optim import os import torchvision.transforms...
1811 просмотров

Как установить PyTorch с помощью pipenv и сохранить его в Pipfile и Pipfile.lock?
В настоящее время я использую Pipenv для поддержки пакетов Python, используемых в конкретном проекте. Большинство загрузок, которые я пробовал до сих пор, работали должным образом; то есть я ввожу pipenv install [package] , и он устанавливает пакет...
993 просмотров
schedule 21.02.2022

Проблемы с PyTorch dll для Caffe2
Я использую компьютер с Windows 10, и после переустановки Anaconda и всех пакетов, которые у меня были ранее, включая torchvision, torch и необходимые зависимости, я все еще получаю эту ошибку: OSError: [WinError 127] Не удалось найти указанную...
1057 просмотров