Укажите размер входа изображения для настройки сети.

imageSize = [224 224 3];

Укажите количество классов объектов, которые должна обнаружить сеть.

numClasses = 1;

Определите анкерные рамки.

anchorBoxes = [1 1;4 6;5 3;9 6];

Укажите предварительно обученную сеть ResNet -50 в качестве базовой сети для YOLO v2. Чтобы использовать эту предварительно обученную сеть, необходимо установить пакет поддержки 'Deep Learning Toolbox Model for ResNet-50 Network'.

network = resnet50();

Проанализируйте сетевую архитектуру, чтобы просмотреть все слои сети.

analyzeNetwork(network)

Укажите слой сети, который будет использоваться для редукции данных. В качестве слоя функции можно выбрать любой слой, кроме полносвязного слоя.

featureLayer = 'activation_49_relu';

Создайте сеть обнаружения объектов YOLO v2. Сеть возвращается как LayerGraph объект.

lgraph = yolov2Layers(imageSize,numClasses,anchorBoxes,network,featureLayer);

Анализ сетевой архитектуры YOLO v2. Слои, следующие за слоем функции, удаляются. Серия слоев свертки, ReLU и нормализации партии . наряду с слоями преобразования YOLO v2 и выхода YOLO v2 добавляются к функции слою основы сети.

analyzeNetwork(lgraph)

Укажите размер входа изображения для настройки сети.

imageSize = [224 224 3];

Укажите количество классов объектов, которые должна обнаружить сеть.

numClasses = 1;

Определите анкерные рамки.

anchorBoxes = [1 1;4 6;5 3;9 6];

Укажите предварительно обученный ResNet -50 как базовую сеть для YOLO v2. Чтобы использовать эту предварительно обученную сеть, необходимо установить пакет поддержки 'Deep Learning Toolbox Model for ResNet-50 Network'.

network = resnet50();

Проанализируйте сетевую архитектуру, чтобы просмотреть все слои сети.

analyzeNetwork(network)

Укажите слой сети, который будет использоваться для редукции данных. В качестве слоя функции можно выбрать любой слой, кроме полносвязного слоя.

featureLayer = 'activation_49_relu';

Укажите слой сети, который будет использоваться в качестве источника для уровня реорганизации.

reorgLayer = 'activation_47_relu';

Создайте сеть обнаружения объектов YOLO v2. Сеть возвращается как LayerGraph объект.

lgraph = yolov2Layers(imageSize,numClasses,anchorBoxes,network,featureLayer,'ReorglayerSource',reorgLayer);

Анализ сетевой архитектуры YOLO v2. Слои, следующие за слоем функции, удаляются. Подсеть обнаружения наряду с преобразованием YOLO v2 и выходными слоями YOLO v2 добавляются к слою функций базовой сети. Слой реорганизации и слой конкатенации глубин также добавляются к сети. Слой перестройки YOLO v2 реорганизовывает размерность выхода функций из activation_47_relu слой. Слой конкатенации глубин конкатенирует выход слоя реорганизации с выходом более высокого слоя.

analyzeNetwork(lgraph)