GPU Coder™ поддерживает генерацию кода для ряда и направленного графа без петель (DAG) сверточные нейронные сети (CNNs или ConvNets). Можно сгенерировать код для любой обученной сверточной нейронной сети, слои которой поддерживаются для генерации кода. Смотрите Поддерживаемые Слои. Можно обучить сверточную нейронную сеть или на центральном процессоре, графическом процессоре или на нескольких графических процессорах при помощи Deep Learning Toolbox™ или использовать одну из перечисленных в таблице предварительно обученных сетей и сгенерировать код CUDA®.
Сетевое имя | Описание | cuDNN | TensorRT |
---|---|---|---|
AlexNet | AlexNet сверточная нейронная сеть. Для предварительно обученной модели AlexNet смотрите | Да | Да |
GoogLeNet | GoogLeNet сверточная нейронная сеть. Для предварительно обученной модели GoogLeNet смотрите | Да | Да |
Caffe Network | Сверточные модели нейронной сети от Caffe. Для импорта предварительно обученной сети от Caffe смотрите | Да | Да |
Начало-v3 | Начало-v3 сверточная нейронная сеть. Для предварительно обученной модели Начала-v3 смотрите inceptionv3 . | Да | Да |
ResNet | ResNet-50 и сверточные нейронные сети ResNet-101. Для предварительно обученных моделей ResNet смотрите | Да | Да |
SegNet | Мультикласс pixelwise сеть сегментации. Для получения дополнительной информации смотрите | Да | Да |
SqueezeNet | Маленькая глубокая нейронная сеть. Для предварительно обученных моделей ResNet смотрите | Да | Нет |
VGG-16 | VGG-16 сверточная нейронная сеть. Для предварительно обученной модели VGG-16 смотрите | Да | Да |
VGG-19 | VGG-19 сверточная нейронная сеть. Для предварительно обученной модели VGG-19 смотрите | Да | Да |
Следующие слои поддерживаются для генерации кода GPU Coder для целевых библиотек глубокого обучения, заданных в таблице.
Если вы устанавливаете Интерфейс GPU Coder пакета поддержки для Библиотек Глубокого обучения, можно использовать coder.getDeepLearningLayers
, чтобы видеть список слоев, поддержанных для определенной библиотеки глубокого обучения. Например, coder.getDeepLearningLayers('cudnn')
показывает список слоев, поддержанных для генерации кода при помощи библиотеки NVIDIA® cuDNN.
Имя слоя | Описание | cuDNN | TensorRT |
---|---|---|---|
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 для целевых библиотек глубокого обучения, заданных в таблице.
Имя слоя | Описание | cuDNN | TensorRT |
---|---|---|---|
yolov2ObjectDetector | Обнаружьте объекты с помощью детектора объекта YOLO v2
| Да | Да |