Описание тега torchvision
Используйте этот тег для вопросов, концепций и проблем, связанных с библиотекой компьютерного зрения PyTorch.
2
ответа
Как я могу генерировать и отображать сетку изображений в PyTorch с помощью plt.imshow и torchvision.utils.make_grid?
Я пытаюсь понять, как torchvision взаимодействует с mathplotlib для создания сетки изображений. Легко генерировать изображения и отображать их итеративно: import torch import torchvision import matplotlib.pyplot as plt w = torch.randn(10,3,640,640) …
13 июл '18 в 16:13
2
ответа
Поезд SqueezeNet модель с использованием набора данных MNIST Pytorch
Я хочу обучить модель SqueezeNet 1.1, используя набор данных MNIST вместо набора данных ImageNet. Могу ли я иметь ту же модель, что и torchvision.models.squeezenet? Спасибо!
03 дек '18 в 11:56
2
ответа
Как использовать AlexNet с одним каналом
Я новичок в pytorch и имел проблемы с каналами в AlexNet. Я использую его для проекта "gta san andreas self car car", я собрал набор данных из черно-белого изображения с одним каналом и пытаюсь обучить AlexNet с помощью сценария: from AlexNetPytorch…
01 янв '19 в 22:57
2
ответа
Оси не соответствуют массиву в pytorch
Я новичок в Pytorch, и я застрял в этом более недели. Я пытаюсь использовать AlexNet для создания автомобиля с самостоятельным вождением "gta san Andreas", и у меня много проблем с подготовкой данных. сейчас я получаю эту ошибку. Traceback (most rec…
01 янв '19 в 13:10
0
ответов
Fastai 1.0.x использует модель torchvision
Может кто-нибудь показать мне пример использования модели torchvision в fastai без изменений (т.е. без изменений сети, выполненных create_cnn)? Я пробовал это, но получаю RuntimeError: size mismatch, Я думаю, что входной размер alexnet должен быть 2…
21 фев '19 в 20:40
1
ответ
Во время работы программы в Ubuntu 18.04 (работает внутри VMWare) я получаю эту ошибку: AssertionError: Факел не скомпилирован с включенной CUDA
Несмотря на то, что я успешно установил torchvision и cuda, я получаю эту ошибку: Я запускаю программу внутри VMWare (Ubuntu 18.04). python3 ./decode/train_dataloader.py --img_data ./datasets/image_fmri --output ./tmp/feat_data Traceback (most recen…
04 фев '19 в 17:36
0
ответов
RuntimeError: Работник DataLoader уничтожается по сигналу: недопустимая инструкция.
Я использую следующий код. Я запускаю это на компьютере с графическим процессором с нижеуказанным config: nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2017 NVIDIA Corporation Built on Fri_Sep__1_21:08:03_CDT_2017 Cuda compilation tools, …
23 ноя '18 в 16:13
1
ответ
Pytorch DataLoader - Выберите набор классов STL10
Можно ли тянуть только там, где class = 0 в наборе данных STL10 в PyTorch torchvision? Я могу проверить их в цикле, но мне нужно получать партии изображений класса 0 # STL10 dataset train_dataset = torchvision.datasets.STL10(root='./data/', transfor…
14 июл '18 в 02:56
3
ответа
Torchvision 0.2.1 transforms. Нормализация не работает, как ожидалось
Я пытаюсь новый код, используя Pytorch. В этом коде для загрузки набора данных (CIFAR10) я использую наборы данных torchvision. Я определяю две функции преобразования ToTensor() и Normalize(). Я предполагаю, что после нормализации данные в наборе да…
16 ноя '18 в 06:37
1
ответ
Добавить канал в MNIST через преобразование?
Я пытаюсь использовать набор данных MNIST из torchvision.datasets. Похоже, что предоставляется как N x H x W (uint8) (размер партии, высота, ширина) тензор. Все классы pytorch для работы с изображениями (например, Conv2d) однако требуют N x C x H x …
15 фев '19 в 10:21
2
ответа
PyTorch - неправильная маркировка с использованием torchvision.datasets.ImageFolder
Я структурировал свой набор данных следующим образом: dataset/train/0/456.jpg dataset/train/1/456456.jpg dataset/train/2/456.jpg dataset/train/... dataset/val/0/878.jpg dataset/val/1/234.jpg dataset/val/2/34554.jpg dataset/val/... Так что я использо…
11 дек '18 в 21:02
2
ответа
Как поместить наборы данных, созданные torchvision.datasets в GPU, в одну операцию?
Я имею дело с CIFAR10 и использую torchvision.datasets для его создания. Мне нужен графический процессор для ускорения вычислений, но я не могу найти способ поместить весь набор данных в графический процессор за один раз. В моей модели нужно использ…
14 янв '19 в 01:38
0
ответов
Как изменить размер дополненного ввода на канал в Pytorch?
Я пытаюсь настроить классификатор изображений с помощью Pytorch. Мои образцы изображений имеют 4 канала и имеют размер 28x28 пикселей. Я пытаюсь использовать встроенный torchvision.models.inception_v3() в качестве моей модели. Всякий раз, когда я пы…
22 ноя '18 в 21:26
0
ответов
Визуализация графиков тензорной доски с помощью PyTorch
Я пытаюсь визуализировать график модели PyTorch, используя: https://github.com/lanpa/tensorboardX Однако, когда я следую инструкциям на странице GitHub README для демонстрационного примера: Запустите демонстрационный скрипт: python demo.py Используй…
17 авг '18 в 12:07
1
ответ
Учитывая 4 балла, как обрезать четырехугольник из изображения в pytorch/torchvision?
Проблема довольно проста - Я хочу обрезать четырехугольник из изображения в pytorch/torchvision. Учитывая, у меня есть четыре координаты углов этого четырехугольника. Обратите внимание, что эти четыре точки ограничивают четырехугольник внутри себя, …
29 сен '18 в 07:10
1
ответ
Передача обучения PyTorch с предварительно обученной моделью ImageNet
Я хочу создать классификатор изображений, используя трансферное обучение по модели, уже обученной в ImageNet. Как заменить последний слой torchvision.models Классификатор ImageNet с моим собственным классификатором?
21 дек '18 в 12:18
2
ответа
ImportError: нет модуля с именем "torchvision.datasets.mnist"
Даже после установки pytorch эта ошибка появляется для этой строки. from torchvision import datasets
20 янв '19 в 08:22
1
ответ
ПИТОРЧ ФАКЕЛ ВИДЕНИЕ МНИСТ скачать
Я новичок в Pytorch и Torchvision. Я следовал учебному пособию, которому примерно год, и он пытался скачать mnist через python и torchvision. Вот как: import torch from torchvision import datasets, transforms kwargs = {'num_workers': 1, 'pin_memory'…
01 мар '19 в 18:35
2
ответа
Как ускорить "ImageFolder" для ImageNet
Я нахожусь в университете, и все файловые системы находятся в удаленной системе, где бы я ни входил со своей учетной записью, я всегда мог получить доступ к своему домашнему каталогу. хотя я вхожу на серверы GPU через команду SSH. Это условие, при к…
15 янв '19 в 21:37
1
ответ
Несоответствие размеров, m1: [3584 x 28], м2: [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 = transfor…
16 янв '19 в 13:53