为YOLO v2对象检测网络创建输出层
的yolov2OutputLayer
函数创建YOLOv2OutputLayer
对象,它代表输出层,供您只看一次版本2 (YOLO v2)对象检测网络。输出层提供目标对象的细化边界框位置。
创建一个层
= yolov2OutputLayer (anchorBoxes
)YOLOv2OutputLayer
对象,层
,表示YOLO v2对象检测网络的输出层。该层输出精炼的边界框位置,这些位置是使用输入处指定的一组预定义锚框预测的。
使用名称-值对和前面语法中的输入设置附加属性。将每个属性名用单引号括起来。例如,层
= yolov2OutputLayer (anchorBoxes
,名称,值
)yolov2OutputLayer('名称',' yolo_Out ')
创建一个名为“yolo_Out”的输出层。
为了提高预测精度,您可以:
用更多的图像训练网络。您可以通过数据增强来扩展训练数据集。有关如何将数据增强应用于训练数据集的信息,请参见用于深度学习的图像预处理(深度学习工具箱).
进行多尺度训练trainYOLOv2ObjectDetector
函数。为此,指定'TrainingImageSize
的论证trainYOLOv2ObjectDetector
网络训练功能。
选择适合于数据集的锚框来训练网络。您可以使用estimateAnchorBoxes
函数直接从训练数据计算锚盒。
[1]约瑟。R、S. K.迪夫瓦拉、R. B.格什克和F.阿里。“你只看一次:统一、实时的物体检测。”在IEEE计算机视觉与模式识别会议论文集(CVPR),第779-788页。拉斯维加斯,内华达州:CVPR, 2016。
[2]约瑟。R和f阿里。“YOLO 9000:更好、更快、更强。”在IEEE计算机视觉与模式识别会议论文集(CVPR)第6517-6525页。檀香山,HI: CVPR, 2017。