Exponenta Banner
exponenta event banner

importCaffeLayers

Импорт сверточных слоев нейронной сети из Caffe

свернуть все на странице

Синтаксис

layers = importCaffeLayers (протофайл)
layers = importCaffeLayers (протофайл, InputSize, sz)

Описание

пример

layers = importCaffeLayers(protofile) импортирует уровни сети Caffe [1]. Функция возвращает слои, определенные в .prototxt файл protofile.

Для этой функции требуется пакет поддержки Deep Learning Toolbox™ Importer для моделей Caffe. Если этот пакет поддержки не установлен, функция предоставляет ссылку для загрузки.

Предварительно подготовленные сети можно загрузить из зоопарка Caffe Model [2].

layers = importCaffeLayers(protofile,'InputSize',sz) определяет размер входных данных. Если .prototxt в файле не указан размер входных данных, затем необходимо указать размер входных данных.

Примеры

свернуть все

В этом примере используются:

Загрузите и установите пакет поддержки Deep Learning Toolbox Importer для моделей Caffe.

Загрузите необходимый пакет поддержки путем ввода importCaffeLayers в командной строке.

importCaffeLayers

Если пакет поддержки Deep Learning Toolbox Importer for Caffe Models не установлен, функция предоставляет ссылку на необходимый пакет поддержки в проводнике Add-On. Чтобы установить пакет поддержки, щелкните ссылку и нажмите кнопку Установить.

В этом примере используются:

Укажите файл примера 'digitsnet.prototxt' для импорта.

protofile = 'digitsnet.prototxt';

Импортируйте сетевые слои.

layers = importCaffeLayers(protofile)
layers = 

  1x7 Layer array with layers:

     1   'testdata'   Image Input             28x28x1 images
     2   'conv1'      Convolution             20 5x5x1 convolutions with stride [1  1] and padding [0  0]
     3   'relu1'      ReLU                    ReLU
     4   'pool1'      Max Pooling             2x2 max pooling with stride [2  2] and padding [0  0]
     5   'ip1'        Fully Connected         10 fully connected layer
     6   'loss'       Softmax                 softmax
     7   'output'     Classification Output   crossentropyex with 'class1', 'class2', and 8 other classes

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

свернуть все

Имя файла .prototxt файл, содержащий архитектуру сети, указанный как вектор символов или строковый скаляр. protofile должен находиться в текущей папке, в папке на пути MATLAB ®, либо необходимо указать полный или относительный путь к файлу. Если .prototxt в файле не указан размер входных данных, необходимо указать размер с помощью sz входной аргумент.

Пример: 'digitsnet.prototxt'

Размер входных данных, указанный как вектор строки. Укажите вектор из двух или трех целых значений [h,w], или [h,w,c] соответствует высоте, ширине и количеству каналов входных данных.

Пример: [28 28 1]

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

свернуть все

Сетевая архитектура, возвращенная как Layer массив или LayerGraph объект. Сети Caffe, принимающие цветные изображения в качестве входных данных, ожидают, что изображения будут в формате BGR. Во время импорта, importCaffeLayers изменяет сеть таким образом, что импортированная сеть MATLAB принимает изображения RGB в качестве входных данных.

Совет

Ссылки

[2] Зоопарк модели Caffe. https://caffe.berkeleyvision.org/model_zoo.html.

См. также

| | | | | | | |

Темы

Представлен в R2017a

Документация по инструментам для глубокого обучения

Поддержка