为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. Divvala, R. B. Girshick和F. Ali。“你只看一次:统一的、实时的物体检测。”在IEEE计算机视觉与模式识别会议论文集, 779 - 788页。拉斯维加斯,内华达州:CVPR, 2016。
[2]约瑟。R和f。“YOLO 9000:更好、更快、更强。”在IEEE计算机视觉与模式识别会议论文集, 6517 - 6525页。檀香山,HI: CVPR, 2017。