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'

Типы данных: char

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

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

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

свернуть все

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

Советы

  • importCaffeLayers может импортировать сети со следующими типами слоя Caffe с некоторыми ограничениями:

    • Входной параметр

    • Свертка

    • Развертка

    • Пакетная норма

    • ReLU

    • Tanh

    • Локальная нормализация ответа (LRN)

    • Объединение

    • Скалярное произведение / Полностью Связанный

    • Уволенный

    • Concat

    • Eltwise (только сумма)

    • Softmax с потерей

    • Евклидова потеря

    Если сеть содержит какой-либо другой тип слоя, то программное обеспечение возвращает ошибку.

    Функция импортирует только слои, которые protofile задает с ТЕСТОМ включать-фазы. Функция игнорирует любые слои, которые protofile задает с TRAIN включать-фазы.

Ссылки

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

Введенный в R2017a