Публикации по теме 'torchvision'


Преобразование модели классификации изображений torchvision в CoreML
Как преобразовать предварительно обученную модель в модель классификации изображений CoreML для использования на iOS Мы хотим преобразовать модель классификации изображений Если вы преобразуете модель классификации изображений TorchVision как есть, на выходе будет числовое значение 1000. Мы хотели бы преобразовать его в процент достоверности, чтобы его можно было предварительно просмотреть как модель классификации изображений. Добавить имя класса и слой softmax Добавьте имя класса..
Понимание функций Torchvision для PyTorch - Часть 2 - Преобразования
Интуитивное понимание библиотеки torchvision - с 14 визуальными примерами преобразований (Часть 2/3) Для Части 1 (введение в модули в torchvision ) перейдите по ссылке ниже. Понимание функций Torchvision (для PyTorch) Интуитивное понимание библиотеки torchvision - от основ до продвинутого (Часть 1/3) medium.com В этой статье Мы будем экспериментировать с некоторые базовые преобразования изображений при загрузке набора..

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

У меня возникла ошибка импорта, когда я попытался импортировать torchvision
Когда я попытался импортировать torchvision, у меня возникла ошибка импорта. Я погуглил эту проблему, но ничего не нашел, кроме того, что проблема может быть в Cython. Поиск StackOverflow тоже ничего не нашел. Как я могу это решить?...
3669 просмотров
python-3.x torchvision
schedule 29.05.2022
PyTorch — получение «TypeError: pic должно быть изображением PIL или ndarray. Получил ошибку класса 'numpy.ndarray'›'
Я получаю сообщение об ошибке TypeError: pic should be PIL Image or ndarray. Got <class 'numpy.ndarray'> при попытке загрузить набор данных без изображений через файл DataLoader . Версии torch и torchvision — это 1.0.1 и 0.2.2.post3...
19839 просмотров
python pytorch deep-learning torch torchvision
schedule 05.02.2023
Как закодировать изображение RGB в горячий тензор n_class One в PyTorch
Итак, я работаю над задачей сегментации, и мне нужно преобразовать изображение RGB в n-канальную горячую матрицу, которая будет использоваться в качестве меток для модели U-Net. Я рассчитал цвета изображений. Число или цвета эквивалентны...
878 просмотров
pytorch computer-vision deep-learning torch torchvision
schedule 29.06.2022
RuntimeError: Ожидается 4-мерный ввод для 4-мерного веса 32 3 3, но вместо этого получен 3-мерный ввод размера [3, 224, 224]?
Я пытаюсь использовать предварительно обученную модель. Вот где возникает проблема Разве модель не должна принимать простое цветное изображение? Почему ожидается четырехмерный ввод? RuntimeError Traceback (most...
37655 просмотров
pytorch machine-learning computer-vision conv-neural-network torchvision
schedule 25.04.2023
Pytorch загружает изображения и наземные изображения для обучения нейронной сети
Я пытаюсь использовать пользовательские данные изображения для обучения и тестирования нейронной сети pytorch. У меня есть папка с изображениями в качестве данных и другая папка с изображениями в качестве истины. Файлы и наземные изображения имеют...
428 просмотров
pytorch neural-network torchvision
schedule 08.09.2022
Исправлено случайное семя для преобразований torchvision.
Я использую код, похожий на следующий, для увеличения данных: from torchvision import transforms #... augmentation = transforms.Compose([ transforms.RandomApply([ transforms.RandomRotation([-30, 30]) ],...
1928 просмотров
python pytorch deep-learning neural-network torchvision
schedule 05.10.2022
Torchvision.transforms реализация Flatten()
У меня есть изображения в градациях серого, но мне нужно преобразовать их в набор данных 1d-векторов. Как мне это сделать? Я не смог найти подходящий метод в преобразованиях: train_dataset =...
1088 просмотров
python pytorch flatten torchvision
schedule 31.01.2024
система показывает, что все запрошенные пакеты уже установлены. однако я получаю сообщение об ошибке при импорте pytorch
Я установил pytorch, используя следующую команду, как указано на их сайте: conda install pytorch torchvision cudatoolkit=10.2 -c pytorch . Однако я попытался использовать горелку для импорта и получил следующую ошибку: 'Traceback (most recent...
703 просмотров
python pytorch torchvision
schedule 28.05.2024
Невозможно импортировать torchvision --AttributeError: модуль torch.jit не имеет атрибута unused. Windows10, Python 3.7
Я пытаюсь импортировать torchvision, но получаю сообщение об ошибке. Я исследовал эту тему в Интернете, и, похоже, возникла проблема с более поздними версиями. Были предложены различные решения - в основном переустановка модулей - которые я пробовал...
4841 просмотров
python-3.x torchvision
schedule 08.06.2022
массив pytorch 4d numpy, применяющий преобразования внутри пользовательского набора данных
Внутри моего пользовательского набора данных я хочу применить transforms.Compose() к массиву NumPy. Мои изображения имеют формат массива NumPy с формой (num_samples, width, height, channels) . Как я могу применить следующие преобразования к...
441 просмотров
pytorch torchvision
schedule 24.02.2023
Python, набор данных класса, как объединить изображения с соответствующими метками в pytorch
Я новичок в PyTorch, и последние пару дней я боролся с классом Dataset, который позволяет вам создавать свой собственный набор данных. Я работаю с этим набором данных ( https://www.kaggle.com/ianmoone0617/flower-goggle-tpu-classification/kernels ),...
593 просмотров
python pytorch dataset torch torchvision
schedule 10.12.2022
RuntimeError: ожидаемый объект скалярного типа Double, но получил скалярный тип Float для аргумента # 2 'other' при вызове _th_max
Файл /home/jake/venv/lib/python3.7/site-packages/torchvision/models/detection/rpn.py, строка 274, в assign_targets_to_anchors match_quality_matrix = self.box_similarity (gt_boxes, anchors_per_image) File / home / jav...
226 просмотров
python pytorch torch torchvision tensor
schedule 06.12.2022
Как использовать torchaudio с torch_xla в google colab TPU?
Я использовал Google Colab (с включенным графическим процессором) для обучения моей модели автоматического распознавания речи на основе pytorch и torchaudio. Но когда я попытался использовать google colab TPU, при обучении моей модели я получил...
700 просмотров
pytorch google-cloud-platform torchvision tpu
schedule 21.07.2023
Как использовать torch.optim.lr_scheduler.OneCycleLR ()?
Я хотел использовать torch.optim.lr_scheduler.OneCycleLR() во время тренировки. Может ли кто-нибудь объяснить мне, как им пользоваться? Из документации я понял, что он должен вызываться после каждого train_batch. Мои заблуждения заключаются в...
858 просмотров
python-3.x pytorch deep-learning torchvision
schedule 15.01.2022
Pytorch: объясните torch.argmax
Здравствуйте, у меня есть следующий код: import torch x = torch.zeros(1,8,4,576) # create a 4 dimensional tensor x[0,4,2,333] = 1.0 # put on 1 on a random spot # I want to find the index of the highest value (0,4,2,333) print(x.argmax()) # this...
235 просмотров
pytorch torch torchvision
schedule 19.04.2022
torch.save () дает: RuntimeError: ошибка CUDA: устройство с поддержкой CUDA не обнаружено
I was training a neural network model on GPU but I get the above mentioned error when I use torch.save() to save checkpoints. My question is even though I have a CUDA device why am I getting the mentioned error? My model was running okay on the GPU...
175 просмотров
pytorch deep-learning torchvision
schedule 20.08.2023
Извлечение выходных данных промежуточного уровня CNN в PyTorch
Я использую модель Resnet18 . ResNet( (conv1): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False) (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (relu): ReLU(inplace=True)...
1890 просмотров
python-3.x python pytorch deep-learning torchvision
schedule 27.07.2022
Pytorch CUDA OutOfMemory Ошибка при обучении
Я пытаюсь обучить модель PyTorch FLAIR в AWS Sagemaker. При этом возникает следующая ошибка: RuntimeError: CUDA out of memory. Tried to allocate 84.00 MiB (GPU 0; 11.17 GiB total capacity; 9.29 GiB already allocated; 7.31 MiB free; 10.80 GiB...
550 просмотров
python pytorch amazon-sagemaker torch torchvision
schedule 02.12.2022
ОШИБКА сборки Docker: не удалось найти версию, удовлетворяющую требованию torch == 1.5.1
У меня есть Dockerfile следующего формата: FROM python:latest ADD requirements.txt . RUN pip3 install -r requirements.txt CMD ["python", "script.py"] Где мой файл requirements.txt выглядит так: grpcio==1.31.0...
418 просмотров
pytorch docker dockerfile torch torchvision
schedule 23.02.2023
5,51 ГиБ уже выделено; 417.00 МиБ бесплатно; 5,53 ГиБ, всего зарезервировано PyTorch CUDA из-за нехватки памяти
Я не уверен, почему запуск этой ячейки вызывает ошибку CUDA out of memory и как ее исправить? Каждый раз мне приходится делать kill -9 PID из списка $ nvidi-smi jupyter notebook, из которого эта ячейка запускается. И все же после нового старта...
721 просмотров
python pytorch torch torchvision
schedule 19.01.2023