车辆目标检测之YOLO算法

时间:2024-03-14 08:55:49

 

这是一个利用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
输出视频大小

车辆目标检测之YOLO算法