coder.DeepLearningConfig

Создайте объекты настройки генерации кода глубокого обучения

Описание

пример

deepLearningCfg = coder.DeepLearningConfig(targetlib)создает объект настройки глубокого обучения, содержащий специфичные для библиотеки параметры что codegen использование, чтобы сгенерировать код для глубоких нейронных сетей. Присвойте этот объект настройки глубокого обучения DeepLearningConfig свойство объекта настройки кода создается при помощи coder.config. Передайте объект настройки кода codegen функция при помощи -config опция.

Примеры

свернуть все

Установите параметры конфигурации кода и сгенерируйте Код С++ для ResNet-50 серийная сеть. Сгенерированный код пользуется библиотеками глубокого обучения Intel® MKL-DNN.

Создайте функцию точки входа resnet_predict это использует coder.loadDeepLearningNetwork функционируйте, чтобы загрузить resnet50 SeriesNetwork объект.

function out = resnet_predict(in)

persistent mynet;
if isempty(mynet)
    mynet = coder.loadDeepLearningNetwork('resnet50', 'myresnet');
end

out = predict(mynet,in);

Постоянный объект старается не восстанавливать и перезагружать сетевой объект во время последующих вызовов функции, чтобы вызвать predict метод на входе.

Входной слой предварительно обученного ResNet-50 сеть принимает изображения размера 224x224x3. Чтобы считать вход отображают от графического файла и изменить размер его к 224x224, используйте следующие строки кода:

in = imread('peppers.png');
in = imresize(in,[224,224]);

Создайте coder.config объект настройки для генерации кода MEX и набора выходной язык на C++. На объекте настройки, набор DeepLearningConfig с targetlib как 'mkldnn'. Используйте -config опция codegen функционируйте, чтобы передать этот объект настройки кода. codegen функция должна определить размер, класс и сложность входных параметров функции MATLAB®. Используйте -args опция, чтобы задать размер входа к функции точки входа.

cfg = coder.config('mex');
cfg.TargetLang = 'C++';
cfg.DeepLearningConfig = coder.DeepLearningConfig('mkldnn'); 
codegen -args {ones(224,224,3,'single')} -config cfg resnet_predict;

codegen команда помещает все сгенерированные файлы в codegen папка. Это содержит Код С++ для функции точки входа resnet_predict.cpp, заголовок и исходные файлы, содержащие определения класса C++ для замысловатой нейронной сети (CNN), веса и файлов смещения.

Входные параметры

свернуть все

Целевая библиотека для генерации кода глубокого обучения, заданной как одно из значений в этой таблице.

ЗначениеОписание
'arm-compute'

Для генерации кода, который пользуется Библиотекой ARM® Compute.

'mkldnn'

Для генерации кода, который использует theIntel Math Kernel Library в Глубоких нейронных сетях (Intel MKL-DNN).

'cudnn'

Для генерации кода, который пользуется библиотекой CUDA® Deep Neural Network (cuDNN).

Эта опция требует GPU Coder™.

'tensorrt'

Для генерации кода, который использует в своих интересах NVIDIA® TensorRT – высокоэффективный оптимизатор вывода глубокого обучения и библиотека времени выполнения.

Эта опция требует GPU Coder.

Выходные аргументы

свернуть все

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

Целевая библиотекаОбъект настройки глубокого обучения
'arm-compute'Создает ARMNEONConfig объект настройки.
'mkldnn'Создает MklDNNConfig объект настройки.
'cudnn'Создает CuDNNConfig объект настройки.
'tensorrt'Создает TensorRTConfig объект настройки.

Введенный в R2018b