主要内容

ssdLayers

SSD多盒对象检测网络

描述

例子

lgraph= ssdLayers (图象尺寸numClassesbaseNetwork创建一个单镜头检测器(SSD)多盒对象检测网络基于baseNetwork,输入图像大小,以及网络应该配置为分类的类的数量。网络被返回为LayerGraph(深度学习工具箱)目的。

SSD是一种基于卷积神经网络的对象检测器,其预测边界框坐标,分类评分和相应的类标签。

lgraph= ssdLayers (___anchorBoxespredictorlayernames.返回包含由指定的自定义锚框的SSDanchorBoxes在指定的位置连接到网络层的predictorlayernames..除了从上一个语法中的输入参数之外,请指定这些参数。

例子

全部收缩

指定基本网络。

baseNetwork ='vgg16'

指定图像大小。

图像= [300 300 3];

指定要检测的类。

numclasses = 2;

创建SSD对象检测网络。

lgraph = ssdLayers(图象尺寸、numClasses baseNetwork);

使用网络分析器可视化网络。

analyzeNetwork (lgraph)

输入参数

全部收缩

输入图像的大小,指定为这些值之一。

  • 形式的两个元素矢量[HW]对于大小的灰度形象H-经过-W

  • 形式的三元素向量[HW3]为RGB彩色图像的大小H-经过-W

当你设置baseNetwork输入'vgg16'“resnet50”,或“resnet101”,图象尺寸输入必须是表单[HW3].

网络要分类的类数,指定为正标量。

预卷曲的卷积神经网络,指定为aLayerGraph(深度学习工具箱)DAGNetwork(深度学习工具箱),或SeriesNetwork(深度学习工具箱)对象或作为这些网络名称之一。要指定其中一个名称,必须下载并安装对应有效网络名称的网络支持包。金宝app

将预先训练好的卷积神经网络作为SSD多盒目标检测网络的基础。有关MATLAB中预训练网络的详细信息®,请参阅普里德深度神经网络(深度学习工具箱)

锚盒,指定为1 by-单元阵列SSD网络中预测器层数。每个预测器层包含一个K-2矩阵定义K形式的锚盒[高度宽度]。每个元件中的锚箱的数量可以变化。

基于输入训练数据中存在的不同对象类的比例和宽高比来确定每个锚盒的大小。每个锚箱的大小必须小于或等于输入图像的大小。您可以使用群集方法来估算训练数据的锚框。有关更多信息,请参阅从训练数据估算锚盒

输入中的图层名称,指定为一个-元素向量或1-by-字符向量的单元格阵列。SSD检测子网连接到该输入指定的预测器层。

输出参数

全部收缩

SSD Multibox对象检测网络,作为一个返回LayerGraph(深度学习工具箱)目的。

请注意

的默认值正常化返回的图像输入层的属性lgraph对象设置为正常化指定的基础网络的属性baseNetwork

算法

ssdLayers函数创建SSD网络并返回lgraph,该对象表示SSD对象检测器的网络架构。

trainssdobjectdetector函数训练并返回SSD对象检测器,ssdObjectDetector.使用检测对象函数ssdObjectDetector对象,使用经过SSD网络架构训练的检测器检测对象。

bbox =检测(探测器,我)

ssdLayers功能以预先训练好的神经网络为基础网络,在基础网络上增加了创建SSD对象检测网络所需的检测子网。给定一个基本网络,ssdLayers删除基础网络中特征层之后的所有层,并添加检测子网。检测子网络由串行连接的卷积层、整流线性单元(ReLU)和批处理归一化层组成。在检测子网络中加入SSD合并层、盒回归层和焦损分类层。

参考

[1] Liu, Wei, Dragomir angelov, Dumitru Erhan, Christian Szegedy, Scott Reed, Cheng-Yang Fu, Alexander C. Berg。“SSD: Single Shot MultiBox Detector”。在计算机视觉- ECCV 2016, Bastian Leibe, Jiri Matas, Nicu Sebe和Max Welling编辑,9905:21-37。Cham:施普林格International Publishing, 2016。https://doi.org/10.1007/978 - 3 - 319 - 46448 - 0 - _2。

[2]黄,乔纳森,Vivek Rathod,陈国,孟隆荣朱,Anoop Korattikara,Alireza Fathi,Ian Fischer,等。“现代卷积对象探测器的速度/准确性权衡。”在2017年电脑视觉和模式识别的IEEE会议(CVPR), 3296 - 97。火奴鲁鲁,你好:IEEE 2017。https / / doi.org/10.1109/CVPR.2017.351。

介绍了R2020a