点击下载项目源代码
整体目录结构如下:

-
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检测模型的整个超参文件。
本文详细介绍了YOLOv3项目的整体目录结构及其各组成部分的功能。包括模型文件、数据集类别文件、锚框配置文件等,并解释了关键模块如模型构建、数据处理及训练过程。

1万+

被折叠的 条评论
为什么被折叠?



