Exponenta Banner

importCaffeLayers

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

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

Синтаксис

layers = importCaffeLayers(protofile)
layers = importCaffeLayers(protofile,'InputSize',sz)

Описание

пример

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

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

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

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

Примеры

свернуть все

Этот пример использует:

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

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

importCaffeLayers

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

Этот пример использует:

Укажите файл примера '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 как входные.

Совет

Ссылки

См. также

| | | | | | | |

Темы

Введенный в R2017a

Документация по Deep Learning Toolbox

Поддержка