Выберите Function to Import ONNX Pretrained Network

Конвертер Deep Learning Toolbox™ для Формата Модели ONNX™ обеспечивает три функции, чтобы импортировать предварительно обученную сеть ONNX (Open Neural Network Exchange): importONNXNetwork, importONNXLayers, и importONNXFunction.

Эта блок-схема иллюстрирует, какая функция импорта лучше всего удовлетворяет различным сценариям.

Flow chart for selecting ONNX import function

Решения

Эта таблица описывает каждое решение в рабочем процессе для выбора функции импорта ONNX.

РешениеОписание
Сеть ONNX имеет несколько выходных параметров?
  • Можно импортировать сеть ONNX с несколькими входными параметрами и одним выходным использованием importONNXNetwork. Если сеть имеет несколько выходных параметров, использовать importONNXLayers или importONNXFunction.

  • importONNXLayers вставляет слои заполнителя для выходных параметров. Для примера смотрите Импорт и Соберите Сеть ONNX с Несколькими Выходными параметрами.

  • importONNXFunction поддержки, импортирующие сети ONNX с несколькими вводами и выводами.

Действительно ли все - поддерживаемые слоя сети ONNX?
  • Если сеть ONNX содержит слой, который не поддерживает Конвертер Deep Learning Toolbox для пакета поддержки Формата Модели ONNX (см. Поддерживаемые Слои ONNX), то importONNXNetwork возвращает сообщение об ошибке.

  • Если сеть ONNX содержит неподдерживаемый слой, то importONNXLayers вставляет слой заполнителя вместо неподдерживаемого слоя.

  • importONNXFunction поддержки большинство слоев ONNX. Для получения дополнительной информации смотрите Поддерживаемые Слои ONNX importONNXFunction.

Вы развернете импортированную сеть?Если вы используете importONNXNetwork или importONNXLayers, можно сгенерировать код для импортированной сети. Создать DAGNetwork объект для генерации кода, смотрите Предварительно обученные сети Загрузки для Генерации кода (MATLAB Coder).
Вы загрузите импортированную сеть с Deep Network Designer?Если вы используете importONNXNetwork или importONNXLayers, можно загрузить импортированную сеть с приложением Deep Network Designer.
Если вы переобучите импортированную сеть, вы будете использовать пользовательский учебный цикл?

Действия

Эта таблица описывает каждое действие в рабочем процессе для выбора функции импорта ONNX.

ДействиеОписание
Использование importONNXNetworkimportONNXNetwork возвращает DAGNetwork объект, который готов использовать для предсказания. Использование classify выполнять предсказания.
Использование importONNXLayersimportONNXLayers возвращает layerGraph объект со слоями заполнителя вместо неподдерживаемых слоев. Найдите и замените слои заполнителя. Соберите использование графика слоев assembleNetwork, который возвращает DAGNetwork объект.
Использование importONNXFunctionimportONNXFunction возвращает ONNXParameters объект, который содержит сетевые параметры и функцию модели (см. Импортированную Функцию Модели ONNX), который содержит сетевую архитектуру. ONNXParameters возразите и функция модели готовы использовать для предсказания. Для примера смотрите, Предсказывают Используя Импортированную Функцию ONNX.
Найдите и замените слои заполнителяЧтобы найти имена и индексы неподдерживаемых слоев в сети, используйте findPlaceholderLayers функция. Затем можно заменить слой заполнителя на новый слой, который вы задаете. Чтобы заменить слой, использовать replaceLayer.

Смотрите также

| | | | |

Похожие темы