Exponenta Banner

importCaffeLayers

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

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

Синтаксис

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

Описание

пример

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

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

Можно загрузить предварительно обученные сети с Зоопарка Модели Caffe [2].

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

Примеры

свернуть все

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

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

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

importCaffeLayers

Если Средство импорта Deep Learning Toolbox для пакета поддержки Моделей Caffe не установлено, то функция обеспечивает ссылку на необходимый пакет поддержки в 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 в качестве входа.

Советы

Ссылки

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

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

| | | | | |

Темы

Введенный в R2017a

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

Поддержка