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

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

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

Сетевое имяОписаниеcuDNNTensorRT
AlexNet

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

ДаДа
GoogLeNet

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

ДаДа
Caffe Network

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

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

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

ДаДа
SegNet

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

ДаДа
SqueezeNet

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

ДаНет
VGG-16

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

ДаДа
VGG-19

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

ДаДа

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

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

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

Имя слояОписаниеcuDNNTensorRT
additionLayer

Слой Addition

ДаДа
averagePooling2dLayer

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

ДаДа
batchNormalizationLayer

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

ДаДа
classificationLayer

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

ДаДа
clippedReluLayer

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

ДаНет
convolution2dLayer

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

ДаДа
crop2dLayer

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

ДаДа
crossChannelNormalizationLayer

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

ДаДа
depthConcatenationLayer

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

ДаДа
dropoutLayer

Слой Dropout

ДаДа
fullyConnectedLayer

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

ДаДа
imageInputLayer

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

ДаДа
leakyReluLayer

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

ДаДа
maxPooling2dLayer

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

ДаДа
maxUnpooling2dLayer

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

ДаДа
PixelClassificationLayer

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

ДаДа
regressionLayer

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

ДаДа
reluLayer

Слой Rectified Linear Unit (ReLU)

ДаДа
softmaxLayer

Слой Softmax

ДаДа
transposedConv2dLayer

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

ДаДа
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 входа.

ДаДа

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

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

Имя слояОписаниеcuDNNTensorRT
yolov2ObjectDetector

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

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

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

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

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

ДаДа

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

Похожие темы