文章目录
前言
目标检测
一、yolov3是什么?
yoloV3是实时目标检测算法yolo的第三个版本,其本身基于darknet构建的神经网络算法.
二、下载源代码及安装依赖库
1.下载keras-yolov3源码
git clone https://github.com/rickkcir/keras-yolo3.git
2.安装pillow库
Pillow 是 PIL 的替代版本,PIL 软件包提供了基本的图像处理功能,如:改变图像大小,旋转图像,图像格式转换,色场空间转换,图像增强,直方图处理,插值和滤波等等。
conda install pillow==5.2.0
3.安装matplotlib库
Matplotlib 是 Python 的绘图库,用于在Python中创建静态,动画和交互式可视化
conda install matplotlib==3.0.0
三、权重文件的下载及转换
1.下载权重文件
wget https://pjreddie.com/media/files/yolov3.weights
2.将权重文件weights转换为.h5文件
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
四、测试
- 测试图片
从百度云盘[密码:1234]下载图片用于测试
python yolo_video.py --image
- 测试视频
2.1 在keras-yolo文件下新建一个output文件夹,用于存放yolo目标检测后的视频。
2.2 从百度云盘[密码:1234]下载测试视频放在keras-yolo文件下。
2.3 执行命令
python yolo_video.py --input yolov3_test.wmv --output output/yolov3_test_result.wmv
keras-yolov3检测视频
总结
在Ubuntu16.04安装annaconda,各个版本:安装keras-2.1.6,tensorflow-gpu-1.9.0,pillow,matplotlib。