利用yolov3 进行目标检测（可自定义检测的类别）

最近在研究车辆检测相关算法，因yolo算法的速度和精度基本均能满足要求，原版的yolo v3算法是检测多类物体，本人是在原来代码的基础上修改为检测单目标车辆，并基于预训练权重在VOC2007数据集上进行训练。
源代码：https://github.com/qqwweee/keras-yolo3
原作者已经给出了训练和测试方法，在此不再赘述。

要实现只检测“car”这一类别，要修改的地方有一下几个：

1、修改yolo.cfg文件
将yolo.cfg文件的classes修改为1，所有的filters=255的值修改为filters=18（计算方式为(classes+5)*3）。（为了不改变yolo.cfg，我是直接将yolo.cfg另存为car.cfg，再做修改）
注意：filters仅仅是Yolo层前的filters修改为18，一共三处。

2、修改类别文件
将coco_class.txt和voc_class.txt文件中的类别全部修改成‘car’ ，或在该目录下新建object_classes.txt文件，写入car。

3、修改voc_annocation.py，生成数据集
将voc_annocation.py 中的classes修改成只有“car”一个，

准备数据集VOC2007, 然后运行voc_annocation.py生成2007_train.txt和2007_test.txt,2007_val.txt文件

4、生成预训练权重文件yolo_weights.h5

下载 yolov3.weights ，放到工程根目录下，cmd下切换到工程目录下，使用命令：

python convert.py -w car.cfg yolov3.weights model_data/yolo_weights.h5

注意，此处的car.cfg 为前面新建的，如果直接在原yolo.cfg上改的，则使用yolo.cfg

5、进行训练
运行 train.py 文件，注意更改训练文件目录(annotation_path)，类别文件目录(classes_path)。
训练完成后会生成一个logs文件夹，里面的000文件夹下会有训练好的权重文件，trained_weights_stage_1.h5 是第一阶段训练得到的（只训练后三层），trained_weights_final.h5是最终得到的训练权重文件（全训练）。

6、测试

修改yolo.py文件，将其中的model_path换成自己生成的trained_weights_final.h5文件的目录， classes_path 改成待检测的目标类文件object_classes.txt.

最后就是加载图片或视频进行测试了。

也可根据需要，对应做上面的修改以满足指定类别物体的检测。
初学者，以上如有不正确之处请指出，谢谢！

利用yolov3 进行目标检测（可自定义检测的类别）相关推荐

智慧交通day02-车流量检测实现12：基于yoloV3的目标检测
在本章节代码编写中,发现之前的代码所处的环境是python3,因此导致了cv2.dnn.readNetFromDarknet()在代码运行中导致了i[0]的获值失败,故总结如下: cv2.dnn.re ...
CV之YOLOv3：基于Tensorflow框架利用YOLOv3算法对热播新剧《庆余年》实现目标检测
CV之YOLOv3:基于Tensorflow框架利用YOLOv3算法对热播新剧<庆余年>实现目标检测目录搭建 1.下载代码 2.安装依赖库 3.导出COCO权重解压到checkpoin ...
从YOLOv1到YOLOv3，目标检测的进化之路
本文来自 CSDN 网站,作者 EasonApp. 作者专栏: http://dwz.cn/7ZGrif YOLOv1 这是继 RCNN,fast-RCNN 和 faster-RCNN之后,Ross ...
yolov3为什么对大目标检测不好_基于改进Yolov3的目标检测的研究
晏世武罗金良严庆摘要:目标检测在视频监控.无人驾驶系统.机械自动化等领域起着重要作用.在如今大数据的背景下,为进一步提高Yolov3在不同数据集下的性能,本文以KITTI数据集为基础,利用重新調 ...
【教程】opencv-python+yolov3实现目标检测
目录 [教程]opencv-python+yolov3实现目标检测目标检测概况目标检测是? 目标检测算法? yolov3模型简介性能介绍架构介绍 opencv-python实现 why ope ...
yolov3为什么对大目标检测不好_从YOLOv1到YOLOv3，目标检测的进化之路
引言:如今基于深度学习的目标检测已经逐渐成为自动驾驶,视频监控,机械加工,智能机器人等领域的核心技术,而现存的大多数精度高的目标检测算法,速度较慢,无法适应工业界对于目标检测实时性的需求,这时YOLO ...
OpenCV+yolov3实现目标检测(C++,Python)
OpenCV+yolov3实现目标检测(C++,Python) 目标检测算法主要分为两类:一类是基于Region Proposal(候选区域)的算法,如R-CNN系算法(R-CNN,Fast R-CN ...
pascal行人voc_利用Pascal VOC目标检测数据深度学习进行目标检测
利用 Pascal VOC 目标检测数据深度学习进行目标检测穆玉理 [期刊名称] <通讯世界> [年 ( 卷 ), 期] 2018(000)005 [摘要] 随着信息社会的发展 , 尤其 ...
【yolov3】如何使用摄像头进行目标检测——yolov3-pytorch摄像头检测教程
[yolov3]如何使用摄像头进行目标检测--yolov3-pytorch摄像头检测教程原项目地址:https://github.com/eriklindernoren/PyTorch-YOLOv3 ...

利用yolov3 进行目标检测（可自定义检测的类别）

利用yolov3 进行目标检测（可自定义检测的类别）相关推荐

最新文章

热门文章