主要内容

ssdLayers

(待拆卸)SSD多盒对象检测网络

ssdLayers函数将在以后的版本中删除。使用ssdObjectDetector功能,创建SSD对象检测网络。

描述

例子

lgraph= ssdLayers (图象尺寸numClassesnetworkName创建SSD (single shot detector)多盒目标检测网络networkName,输入图像大小,以及应该配置网络进行分类的类的数量。网络返回为LayerGraph(深度学习工具箱)对象。

SSD是一个基于卷积神经网络的对象检测器,可以预测边界框坐标、分类分数和相应的类标签。

lgraph= ssdLayers (图象尺寸numClasses网络anchorBoxespredictorLayerNames返回一个包含自定义锚框的SSDanchorBoxes在指定位置连接到网络层的predictorLayerNames。除了前面语法中的输入参数外,还要指定这些参数。

例子

全部折叠

指定基本网络。

networkName =“vgg16”

指定图像大小。

imageSize = [300 300 3];

指定要检测的类。

numClasses = 2;

创建SSD对象检测网络。

lgraph = ssdLayers(imageSize,numClasses,networkName);

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

analyzeNetwork (lgraph)

输入参数

全部折叠

输入图像的大小,指定为以下值之一。

  • 形式为[的双元素向量HW]的灰度图像大小H——- - - - - -W

  • 形式为[的三元素向量HW[3]用于RGB大小的彩色图像H——- - - - - -W

当你设置baseNetwork输入“vgg16”“resnet50”,或“resnet101”,图象尺寸输入必须为[HW3)。

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

预训练卷积神经网络,指定为aLayerGraph(深度学习工具箱)DAGNetwork(深度学习工具箱),或SeriesNetwork(深度学习工具箱)对象。

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

预训练的卷积神经网络,指定为字符串或字符向量的“vgg16”“resnet50”,或“resnet101”网络。要指定这些名称之一,必须下载并安装相应有效网络名称的网络支持包。金宝app

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

锚框,指定为1 by-单元阵列SSD网络中的预测器层数。每个预测层包含一个K- × 2矩阵定义K格式为[高度宽度]。每个元素中的锚框数量可以有所不同。

每个锚框的大小是根据输入训练数据中存在的不同对象类别的比例和纵横比来确定的。每个锚框的大小必须小于或等于输入图像的大小。您可以使用聚类方法从训练数据中估计锚框。有关更多信息,请参见从训练数据估计锚盒

输入中的层名,指定为-元素向量的字符串或1-字符向量的单元数组。SSD检测子网连接到该输入指定的预测器层。

输出参数

全部折叠

SSD多盒对象检测网络,返回为aLayerGraph(深度学习工具箱)对象。

请注意

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

算法

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

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

bbox = detect(检测器,I)

ssdLayers函数使用预训练的神经网络作为基网络,并在基网络中添加创建SSD对象检测网络所需的检测子网。给定一个基本网络,ssdLayers移除基网络中继特征层之后的所有层,并添加检测子网。检测子网由串联卷积层、整流线性单元(ReLU)层和批归一化层组成。检测子网中增加SSD合并层、盒回归层、焦损分类层。

参考文献

bbb Liu, Wei, Dragomir Anguelov, Dumitru Erhan, Christian Szegedy, Scott Reed,付成阳,Alexander C. Berg。“SSD:单镜头多盒探测器。”在计算机视觉- ECCV 2016,由Bastian Leibe, Jiri Matas, Nicu Sebe和Max Welling编辑,9905:21-37。中国:b施普林格国际出版,2016。https://doi.org/10.1007/978 - 3 - 319 - 46448 - 0 - _2。

[2] Huang, Jonathan, Vivek Rathod,孙晨,朱梦龙,Anoop Korattikara, Alireza Fathi, Ian Fischer,等。“现代卷积对象检测器的速度/精度权衡”。在2017 IEEE计算机视觉与模式识别大会(CVPR), 3296 - 97。檀香山,HI:IEEE, 2017。https / / doi.org/10.1109/CVPR.2017.351。

版本历史

在R2020a中引入

全部展开