ONNX - это открытый формат для представления моделей глубокого обучения и обеспечения взаимодействия между различными фреймворками.
0 ответов

Получить тензоры веса и смещения от модели ONNX

Я пытаюсь получить доступ к весам для каждого слоя из файла squeezenet.onnx. Я могу просматривать их через Netron, но я хочу, чтобы веса были отдельно в файле. Насколько я знаю, в ONNX нет прямого метода доступа к весам по слоям. Есть идеи, как это …
10 авг '18 в 03:12
0 ответов

Проблема с использованием openCV для загрузки сети из ONNX (python/pytorch)

Я пытаюсь загрузить обученную модель.onnx (из алгоритма передачи нейронного стиля) в cv2. Я видел, что есть cv.dnn.readNetFromONNX() функция, но в cv2 такой функции нет. Я не могу импортировать или загрузить opencv как cv, и поэтому не могу загрузит…
09 ноя '18 в 11:15
0 ответов

"обрезать" граф onnx от его констант (инициализаторов)

У меня есть граф / модель onnx, в которой есть большие константы, поэтому загрузка и анализ ее занимает много времени. Могу ли я "удалить" данные с графика, чтобы я мог осмотреть узлы графика без его данных?
06 ноя '18 в 16:41
0 ответов

Как отключить сообщение nvonnxparser::IParser::parseFromFile?

Когда я вызываю nvonnxparser::IParser::parseFromFile, будет показано следующее сообщение. ----- Разбор модели ONNX network.onnx выполнен ---- Я уже изменил многословно, но это не работает. Как я могу скрыть сообщение? Благодарю.
14 фев '19 в 12:06
1 ответ

PyTorch VAE не удается преобразовать в onnx

Я пытаюсь преобразовать PyTorch VAE в onnx, но я получаю: torch.onnx.symbolic.normal does not exist Проблема, кажется, происходит от reparametrize() функция: def reparametrize(self, mu, logvar): std = logvar.mul(0.5).exp_() if self.have_cuda: eps = …
14 фев '19 в 21:19
0 ответов

onnxruntime - экспорт scikit RandomForestClassifier не выполняется при импорте

Я экспортировал модель scikit RandomForestClassifier в ONNX, используя WinMLTools. Когда я пытался импортировать файл, используя ONNXRuntime (на C#). Я получаю следующую ошибку: System.AccessViolationException: 'Attempted to read or write protected …
27 фев '19 в 14:54
0 ответов

Экспорт модели pytorch через onnx в caffe2

Я пытаюсь обновить мою существующую модель pytorch 0.4 до версии 1.0 и пытаюсь использовать бэкэнд Caffe2 для запуска моделей, работающих на GPU. Итак, я сделал следующее: # Export my model to ONNX torch.onnx._export(model, args, "test.onnx", export…
09 янв '19 в 15:45
0 ответов

ValueError: Не знаю, как перевести оп Unsqueeze при запуске преобразованной модели PyTorch

Я сталкиваюсь с проблемами, пытаясь использовать модель PyTorch, экспортированную как модель ONNX с Caffe2. Вот мой экспортный код the_model = torchvision.models.densenet121(pretrained=True) garbage, model_inputs = preprocessing("test.jpg") torch_ou…
07 окт '18 в 01:39
1 ответ

Не удается найти файл ONNX, указанный с помощью WinML

Я пытаюсь загрузить модель ONNX, используя WinML, со следующим кодом: private async Task InitializeModelAsync() { var path = @"ms-appx:///Assets/FER-Emotion-Recognition.onnx"; var file = await StorageFile.GetFileFromApplicationUriAsync(new Uri(path)…
27 июл '18 в 23:24
0 ответов

Понимать шаблон lstm_cell в парсере tf2onnx

Я пытался преобразовать lstm frozen.pb, который нашел в https://github.com/onnx/tensorflow-onnx/tree/master/tests/models/lstm Но парсер tf2onnx не смог распознать это как модель lstm, скорее он воспринимает это как нормальную модель. При попытке отл…
28 окт '18 в 15:10
0 ответов

Как использовать onnx_tf для экспорта модели Keras h5

Я следовал учебнику по ONNX. В конце они экспортируют модель в GraphDef прото: f_rep.export_graph(EXPORT_PATH + 'mnist-test/mnist-tf-export.pb') Есть ли способ взять модель, затем экспортировать и сохранить ее в виде файла Keras .h5 (HDF5)?
07 ноя '18 в 16:05
0 ответов

Как вы генерируете представление ONNX нейронной сети, прошедшей предварительную подготовку?

Я пытаюсь создать файл ONNX для примера run_classifier.py pytorch-pretrained-bert. В этом случае я запускаю его со следующими параметрами согласно основному README.md: export GLUE_DIR=/tmp/glue_data python run_classifier.py \ --task_name MRPC \ --do…
16 янв '19 в 15:12
1 ответ

Передача предварительно обученной модели pytorch в onnx

Я пытаюсь преобразовать модель pytorch в ONNX, чтобы потом использовать ее для TensorRT. Я следовал следующему руководству https://pytorch.org/tutorials/advanced/super_resolution_with_caffe2.html, но мое ядро ​​все время умирает. Это код, который я …
18 янв '19 в 12:45
0 ответов

Когда ONNX решит включить опцию Gather?

Неясный вопрос, я полагаю, но у меня есть модель PyTorch, которую я только что построил, и она не в состоянии преобразовать в CoreML, потому что ONNX добавил Gather ОПС. Полная модель на самом деле представляет собой объединение двух отдельных модел…
19 фев '19 в 04:19
2 ответа

Преобразование тензорного потока в ONNX

В настоящее время я пытаюсь преобразовать сохраненный (и работающий) файл.pb, созданный с помощью этого руководства ( https://github.com/thtrieu/darkflow), в файл onnx. В настоящее время я использую инструменты WinML, но результат преобразования не …
16 янв '19 в 09:57
1 ответ

Модель ONNX с ошибкой субоператора не связывает

Я пытаюсь создать сеанс с моделью Squeezenet ONNX, используя: session = winrt::Windows::AI::MachineLearning::LearningModelSession{ model, winrt::Windows::AI::MachineLearning::LearningModelDevice(deviceKind) }; У меня есть две версии squeezenet. Один…
26 окт '18 в 16:17
0 ответов

Ошибка сегментации при запуске convert-onnx-to-caffe2

Я пытаюсь запустить convert-onnx-to-caffe2, используя onnx-caffe2: https://github.com/onnx/onnx-caffe2 Я получаю ошибку сегментации: Есть идеи?
21 авг '18 в 16:43
1 ответ

Выполнить вывод, используя модель Onnx в Python?

Я пытаюсь проверить правильность моей модели.onnx, и мне нужно выполнить вывод, чтобы проверить вывод для того же. Я знаю, что мы можем запустить проверку на.mlmodel с помощью coremltools в Python - в основном, загрузить модель и ввод и получить про…
21 июн '18 в 19:06
1 ответ

Есть ли способ конвертировать предварительно обученную модель из PyTorch в ONNX?

Я обучил модель StarGAN на своем наборе данных. И мне нужно конвертировать эту модель из.pth(Pytorch) в.pb для использования на студии Android. Я много искал и нашел способы конвертации. Однако все решения не работают в моем случае. Я пробовал на не…
24 окт '18 в 16:19
1 ответ

Windows ML - ONNX - Исключение из HRESULT: 0x88900105

Я читаю эту статью: добавьте немного машинного обучения в свое приложение Windows благодаря WinML и воспроизведите пример приложения. В настоящее время это работает. На портале Azure Custom Vision я создал свою собственную модель видения и экспортир…
28 окт '18 в 07:48