车辆目标检测之YOLO算法

 

这是一个利用yolo算法实现车流识别的系统。

硬件基础是cpui7-7700HQ，显卡英伟达1050TI。
基于win10，python3。
python的安装库要求cv2，深度学习框架tensflow，keras和一些常用机器学习库。

是否选择gpu加速看个人情况，这里使用cuda8.0，cudnn6.0。

 

亲测已经实现

 

权重yolo.h5下载网址

https://download.csdn.net/download/qq_37534839/10275402

代码

https://github.com/TheEighthDay/yolo

觉着不错请点个star，非常感谢(●'◡'●)

 

可以直接对于图片预测车辆yolo_picture.py，参数为输入图片地址和输出图片地址

此外对于视频的车辆预测yolo_video.py,参数如下

VIDEO_PATH
源视频地址

EXTRACT_FOLDER
生成视频存放的位置

EXTRACT_FREQUENCY
帧提取频率，每几帧提取一张分析

FPS
形成视频的每秒帧数

SIZE
输出视频大小