点击下载项目源代码
整体目录结构如下:
-
font
文件夹中存放的是该项目中使用到的字体,如下:
-
model_data
文件夹中存放的是模型文件,如下:
coco_classes.txt
文件中存放COCO
数据集的类别,共80个类别。voc_classes.txt
文件中存放VOC
数据集的类别,共20个类别。yolo_anchors.txt
文件中存放着yolo
模型使用的anchors
(个人习惯叫标注框)的大小,共9个。tiny_yolo_anchors.txt
文件中存放着tiny_yolo
模型使用的标注框的大小,共6个。
-
yolo3
文件夹中存放有3个文件,如下:
__init__.py
是一个空文件,未存放任何代码;model.py
是用keras
构建yolo3
模型的主要模块文件,共有14个函数,如下:
utils.py
是在模型训练时进行数据处理的工具文件,共有4个函数,如下:
-
*_annotation.py
对数据进行转换的文件:coco_annotation.py
把COCO
数据集标注的json
格式文件转换为txt
文件voc_annotation.py
把VOC
数据集标注的xml
格式文件转换为txt
文件
-
convert.py
把原始权重转换为kares
的能读取的原始h5
文件。 -
kmeans.py
输入上面得到的txt
文件,通过聚类得到指定数量的anchors
。 -
train.py
是进行yolov3
模型训练的文件。 -
train_bottleneck.py
是使用自己的数据集重新训练YOLO模型的文件。 -
yolo.py
构建以yolov3
为底层构件的yolo
检测模型。 -
yolo_video.py
使用yolo.py
文件中的yolo
检测模型对视频中的物体进行检测。 -
yolov3.cfg
构建yolov3
检测模型的整个超参文件。 -
yolov3-tiny.cfg
构建yolov3-tiny
检测模型的整个超参文件。