Поддерживаемые сети и слои

Поддерживаемые предварительно обученные сети

GPU Coder™ поддерживает генерацию кода для ряда и сверточных нейронных сетей направленного графа без петель (DAG) (CNNs или ConvNets). Можно сгенерировать код для любой обученной сверточной нейронной сети, слои которой поддерживаются для генерации кода. Смотрите Поддерживаемые Слои. Можно обучить сверточную нейронную сеть или на центральном процессоре, графическом процессоре или на нескольких графических процессорах при помощи Deep Learning Toolbox™ или использовать одну из перечисленных в таблице предварительно обученных сетей и сгенерировать код CUDA®.

Сетевое имяОписаниеcuDNNTensorRTБиблиотека ARM® Compute для Мали графический процессор
AlexNet

Сверточная нейронная сеть AlexNet. Для предварительно обученной модели AlexNet смотрите alexnet.

ДаДаДа
GoogLeNet

Сверточная нейронная сеть GoogLeNet. Для предварительно обученной модели GoogLeNet смотрите googlenet.

ДаДаДа
Caffe Network

Модели сверточной нейронной сети от Caffe. Для импорта предварительно обученной сети от Caffe смотрите importCaffeNetwork.

ДаДаДа
DeepLab v3 +

DeepLab v3 + сверточная нейронная сеть. Для получения дополнительной информации смотрите deeplabv3plusLayers.

ДаДаНет
DenseNet-201

Сверточная нейронная сеть DenseNet-201. Для предварительно обученной модели DenseNet-201 смотрите densenet201.

ДаДаДа
Inception-v3

Сверточная нейронная сеть Inception-v3. Для предварительно обученной модели Inception-v3 смотрите inceptionv3.

ДаДаДа
Mobilenet-v2

Сверточная нейронная сеть MobileNet-v2. Для предварительно обученной модели MobileNet-v2 смотрите mobilenetv2.

ДаДаДа
ResNet

ResNet-18, ResNet-50 и сверточные нейронные сети ResNet-101. Для предварительно обученных моделей ResNet смотрите resnet50, resnet18, и resnet101.

ДаДаДа
SegNet

Мультикласс pixelwise сеть сегментации. Для получения дополнительной информации смотрите segnetLayers.

ДаДаНет
SqueezeNet

Маленькая глубокая нейронная сеть. Для предварительно обученных моделей SqueezeNet смотрите squeezenet.

ДаДаДа
VGG-16

Сверточная нейронная сеть VGG-16. Для предварительно обученной модели VGG-16 смотрите vgg16.

ДаДаДа
VGG-19

Сверточная нейронная сеть VGG-19. Для предварительно обученной модели VGG-19 смотрите vgg19.

ДаДаДа
Xception

Сверточная нейронная сеть Xception. Для предварительно обученной модели Xception смотрите xception.

ДаДаДа
YOLO v2

Вы только смотрите однажды версия 2 основанный на сверточной нейронной сети детектор объектов. Для получения дополнительной информации смотрите yolov2Layers

ДаДаДа

Поддерживаемые слои

Следующие слои поддерживаются для генерации кода GPU Coder для целевых библиотек глубокого обучения, заданных в таблице.

Если вы устанавливаете Интерфейс GPU Coder пакета поддержки для Библиотек Глубокого обучения, можно использовать coder.getDeepLearningLayers видеть список слоев, поддержанных для определенной библиотеки глубокого обучения. Например, coder.getDeepLearningLayers('cudnn') показывает список слоев, поддержанных для генерации кода при помощи библиотеки NVIDIA® cuDNN.

Имя слояОписаниеcuDNNTensorRTARM вычисляет библиотеку для Мали графический процессор
additionLayer

Слой Addition.

ДаДаДа
averagePooling2dLayer

Средний слой объединения.

ДаДаДа
batchNormalizationLayer

Пакетный слой нормализации.

ДаДаДа
classificationLayer

Создайте классификацию выходной слой.

ДаДаДа
clippedReluLayer

Отсеченный слой Rectified Linear Unit (ReLU).

ДаДаДа
concatenationLayer

Слой конкатенации берет входные параметры и конкатенирует их в заданном измерении.

ДаНетНет
convolution2dLayer

2D слой свертки.

ДаДаДа
crop2dLayer

Слой, который применяет 2D обрезку к входу.

ДаДаДа
crossChannelNormalizationLayer

Мудрый каналом локальный слой нормализации ответа.

ДаДаДа
depthConcatenationLayer

Слой конкатенации глубины.

ДаДаДа
dicePixelClassificationLayer

Слой классификации пикселей с помощью обобщенной потери dice в семантической сегментации.

ДаДаДа
dropoutLayer

Слой Dropout.

ДаДаДа
eluLayer

Слой Exponential linear unit (ELU).

ДаДаНет
fullyConnectedLayer

Полносвязный слой.

ДаДаДа
globalAveragePooling2dLayer

Глобальный средний слой объединения для пространственных данных.

ДаДаДа
groupedConvolution2dLayer

2D сгруппированный сверточный слой.

Генерация кода для графического процессора ARM Mali не поддержана для 2D сгруппированного слоя свертки, который имеет NumGroups набор свойств как 'channel-wise' или значение, больше, чем два.

ДаДаДа
imageInputLayer

Отобразите входной слой.

  1. Генерация кода не поддерживает 'Normalization' заданное использование указателя на функцию.

ДаДаДа
leakyReluLayer

Текучий слой Rectified Linear Unit (ReLU).

ДаДаДа
lstmLayer

Слой Long short-term memory (LSTM).

  1. Для генерации кода, StateActivationFunction свойство должно быть установлено в 'tanh'.

  2. Для генерации кода, GateActivationFunction свойство должно быть установлено в 'sigmoid'.

ДаНетНет
maxPooling2dLayer

Слой объединения Max.

ДаДаДа
maxUnpooling2dLayer

Слой необъединения Max.

ДаДаНет
Output Layer

Все выходные слои включая пользовательскую классификацию или регрессию выводят слои, созданные при помощи nnet.layer.ClassificationLayer или nnet.layer.RegressionLayer.

Для примера, показывающего, как задать пользовательскую классификацию выходной слой и задать функцию потерь, смотрите, Задают Пользовательскую Классификацию Выходной Слой (Deep Learning Toolbox).

Для примера, показывающего, как задать пользовательскую регрессию выходной слой и задать функцию потерь, смотрите, Задают Пользовательскую Регрессию Выходной Слой (Deep Learning Toolbox).

ДаДаДа
PixelClassificationLayer

Создайте слой классификации пикселей для семантической сегментации.

ДаДаДа
regressionLayer

Создайте регрессию выходной слой.

ДаДаДа
reluLayer

Слой Rectified Linear Unit (ReLU).

ДаДаДа
sequenceInputLayer

Последовательность ввела слой.

  1. Для генерации кода только поддерживаются векторные входные типы. 2D и 3-D вход последовательности изображений не поддержан.

  2. Генерация кода не поддерживает 'Normalization' заданное использование указателя на функцию.

ДаНетНет
softmaxLayer

Слой Softmax.

ДаДаДа
tanhLayer

Гиперболическая касательная (tanh) слой.

ДаДаДа
transposedConv2dLayer

Создайте транспонированный 2D слой свертки.

Генерация кода не поддерживает асимметричную обрезку входа. Например, задавая векторный [t b l r] для 'Cropping' параметр, чтобы обрезать верхнюю часть, нижнюю часть, оставленную, и право на вход, не поддержан.

ДаДаДа
wordEmbeddingLayer

Слой встраивания слова сопоставляет словари с векторами.

ДаНетНет
YOLOv2OutputLayer

Создайте выходной слой для сети обнаружения объектов YOLO v2.

ДаДаДа
YOLOv2ReorgLayer

Создайте слой перестройки для сети обнаружения объектов YOLO v2.

ДаДаДа
YOLOv2TransformLayer

Создайте преобразовывают слой для сети обнаружения объектов YOLO v2.

ДаДаДа
nnet.keras.layer.FlattenCStyleLayer

Сгладьте активации в 1D C-стиль принятия (упорядоченный по строкам) порядок.

ДаДаДа
nnet.keras.layer.GlobalAveragePooling2dLayer

Глобальный средний слой объединения для пространственных данных.

ДаДаДа
nnet.keras.layer.SigmoidLayer

Сигмоидальный слой активации.

ДаДаДа
nnet.keras.layer.TanhLayer

Гиперболический слой активации касательной.

ДаДаДа
nnet.keras.layer.ZeroPadding2dLayer

Нулевой дополнительный слой для 2D входа.

ДаДаДа
nnet.onnx.layer.ElementwiseAffineLayer

Слой, который выполняет поэлементное масштабирование входа, сопровождаемого сложением.

ДаДаНет
nnet.onnx.layer.FlattenLayer

Сглаживает пространственные размерности входного тензора к размерностям канала.

ДаДаДа
nnet.onnx.layer.IdentityLayer

Слой, который реализует единичный оператор ONNX™.

ДаДаДа

Поддерживаемые классы

Следующие классы поддерживаются для генерации кода GPU Coder для целевых библиотек глубокого обучения, заданных в таблице.

ClassName ОписаниеcuDNNTensorRTARM вычисляет библиотеку для Мали графический процессор
yolov2ObjectDetector

Обнаружьте объекты с помощью детектора объектов YOLO v2

  1. Только detect метод yolov2ObjectDetector поддерживается для генерации кода.

  2. roi аргумент к detect метод должен быть codegen константой (coder.const()) и 1x4 вектор.

  3. Только Threshold, SelectStrongest, MinSize, и MaxSize Пары "имя-значение" поддерживаются.

  4. Метки выход возвращены как массив ячеек из символьных векторов, такой как {'автомобиль', 'шина'}.

ДаДаДа

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

Похожие темы