• Stars
    star
    136
  • Rank 267,670 (Top 6 %)
  • Language
    Python
  • Created about 6 years ago
  • Updated almost 6 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

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

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,但都经过一定归一化处理。x,y是相对于格子的偏移量经过了除以整个图片大小的归一化。

举例说明:

就是原图目标的中心坐标是x1,y1,原图宽高分别为w1,h1假设目标中心坐落在下标为[0,1]的格子里,即int(x1/image_size* cell_size),int(y1/image_size* cell_size)=0,1,此时对应格子的目标置信度应该都为1,x和y应该是相对于[0,1]这个格子的偏移量,具体算法是:x=x1/image_size* cell_size-0,y=y1/image_size* cell_size-1。

w,h也进行归一化但还要开方,具体算法为:w=square(w1/image_size),h=square(h1/image_size),归一化可以把数值指定在一定范围有利于训练,把w,h开方,是因为w,h的值不同于x,y是格子的偏移量限定于一定区间,w,h是针对整个图片而言,变化没那么平缓,所以进行开方。

真实训练数据也按上述方法来处理,只不过刚开始的shape是[cell_size,cell_size,4]然后将它reshape成[cell_size,cell_size,1,4]再复制成[batch,cell_size,cell_size,box_per_cell,4]

关于损失函数计算有目标损失,无目标损失,类别损失,目标框损失,占比不同,实际显示图片要加上非极大值抑制,把两个很相近的目标框只保留置信度高的。

yolo v2

关于yolo v2 网上博客大致内容介绍很详细,可以参考论文YOLO9000: Better, Faster, Stronger
我主要介绍它的训练数据长什么样,这也是困扰我好久的。yolo v2 加了anchor box为上述每个格子提供多个目标的可能,真实值的目标要与anchor box计算iou值,大于阈值才保留,否则保留iou值最大的目标,这样label的shape就变成了[batch, cell_size, cell_size, N_ANCHORS, num_clsses+5],相关坐标x,y,w,h和yolo1处理方式也些许不同,感兴趣的同学可以去参考论文。

代码介绍

代码只针对于yolo1的训练和测试

环境说明:

主要环境配置为:
ubuntu 16.04
python 3.5.5
tensorflow 1.4.1
opencv 3.4.1
不知道windows可不可以,应该没问题

下载数据

训练数据下载VOC解压放置到data目录下,预训练模型放置到data目录下

代码介绍

data下放置训练数据和预训练模型和将数据生成的tfrecords文件
graph保存训练过程中的训练集和验证集的graph
model保存训练的最优model
output是测试图片保存目录
picture是测试图片放置目录
utils包括配置文件config,模型文件model,数据处理文件psscal_voc
image_test.py是判断生成tfrecords文件是否图片标注正确
test.py是测试文件
tfrecord.py是将数据处理成tfrecords格式
train.py是训练文件

运行

首先可以手动修改config配置文件
若要训练的话:
运行python tfrecord.py 生成数据
运行python train.py 训练数据

若要测试:
把自己喜欢图片放到picture内,本代码图片来源于百度图片
查看代码,确定你进行测试要使用的model,运行test.py
本测试代码支持摄像头

建议

建议下载预训练模型训练,训练次数不宜过长,否则过拟合很严重
本代码只保存验证集上的最优模型
代码参考hizhangp
如有错误还请多多指正

结果展示




More Repositories

1

tensorflow-MTCNN

人脸检测MTCNN算法,采用tensorflow框架编写,从理解到训练,中文注释完全,含测试和训练,支持摄像头
Python
365
star
2

tensorflow-facenet

人脸识别算法,结合facenet网络结构和center loss作为损失,基于tensorflow框架,含训练和测试代码,支持从头训练和摄像头测试
Python
152
star
3

DCT-NET.Pytorch

unofficial implementation of DCT-Net: Domain-Calibrated Translation for Portrait
Python
98
star
4

Simple-Lora

diffusion lora chinese tutorial,虚拟idol训练中文教程
Python
95
star
5

tensorflow-deeplab_v3_plus

图像分割算法deeplab_v3+,基于tensorflow,中文注释,摄像头可用
Python
90
star
6

GFPGAN-1024

GFPGAN 1024
Python
37
star
7

HeSer.Pytorch

unofficial implementation of Few-Shot Head Swapping in the Wild
Python
33
star
8

HeadSwap

This is a HeadSwap project not only face
Jupyter Notebook
28
star
9

Wind_predict_seq2seq

基于seq2seq模型的风功率预测
Jupyter Notebook
27
star
10

Learning_Spark

pyspark学习教程
16
star
11

My_Recommendation_System

Recommendation System based on python
Python
11
star
12

LVT

a version tools. face detector,face landmark detector,face parsing and so on
Python
10
star
13

CPM_landmark

基于CPM网络结构加入moblienet思想的关键点检测模型
Python
8
star
14

Triton-Torch-Custom

Triton-Pytorch Custom operator tutorial
Python
5
star
15

hourglass_tensorflow_demo

基于tensorflow的hourglass模型demo
Python
4
star
16

OPENGL-FRAMEWORK-PYTHON

This is a convenient and extensible Python OpenGL framework. You can easily refer to the example shader to realize the custom rendering function
Python
3
star
17

LearnCpp

learn C++ tutorial
C++
2
star
18

My_RL_Learning

Learn RL from MorvanZhou
Python
2
star
19

LearnSwift

swift开发ios教程
Swift
1
star
20

LEARN-SDK

C++
1
star
21

Move-Stone

一些基础模型代码
Python
1
star