您的位置:首页 > 编程语言 > Python开发

YOLO基础教程(一):Python环境搭建与测试

2019-02-28 15:35 369 查看

一、所需库文件

  • Python 3.5+ 开发环境
  • TensorFlow (如果安装GPU版建议安装1.9.0,否则可能会出现错误)
  • OpenCV 3.4.2+

二、下载darkflow

  • 源文件地址:github.com/thtrieu/darkflow  点击download即可
  • 编译darkflow  在cmd中,将路径切换到 darkflow 所在目录,输入指令:python setup.py build_ext --inplace   即可完成编译

编译完后,界面如下:

三、 下载YOLO权重文件

 

视频以YOLO V2为例,所以下载的是打框的文件。(下载V3的权重,发现文件太大,估计后面程序看懂后才能改)

  • 在darkflow 下创建一个名为bin的文件夹,把下载的权重复制进去,在复制一个待检测的视频 test.mp4,最后文件夹如下所示:

四、  测试

在cmd中,将路径切换到 darkflow 所在目录,输入指令:python flow --model cfg/yolo.cfg --load bin/yolo.weights --demo test.mp4 --gpu 1.0 --saveVideo   即可完成编译(没装GPU的将 --gpu 1.0去掉即可,注意指令与自己的文件名称,路径相对应)

运行结果图:

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: