yolo算法python代码_tensorflow-YOLO1

tensorflow-YOLO1

目标检测yolo算法，采用tensorflow框架编写，中文注释完全，含测试和训练，支持摄像头

模型简介

yolo v1

yolo1是端对端的目标检测模型，参考论文为You Only Look Once:Unified, Real-Time Object Detection

主要思想是将图片分割成cell_size * cell_size的格子，每个格子里只包含一个目标，通过网络来输出每个格子的目标值，其中判断格子中是否有目标即判断目标中心点是否在对应格子中。

模型大致结构图如下：

模型经过多层卷积和全连接层，将图片最终输出尺寸为[batch,cell_size * cell_size * (num_classes+ box_per_cell* 5)]。

简单介绍一下输出的表示:

通过reshape分成[batch,cell_size,cell_size,num_classes]表示每个格子对应的类别;

[batch_,cell_size,cell_size,box_per_cell]表示每个格子中是否存在目标的置信度，之所以选择两个box_per_cell是为了让预测精度更准确，通俗来讲就是三个臭皮匠顶一个诸葛亮;

[batch,cell_size,cell_size,box_per_cell,4]表示每个格子每个选框中的目标框的坐标值，4个值分别为目标图片的中心横纵坐标偏移值x,y，目标图片的长宽w,h&#