新手入门,通过该贴记录一下,目的是在win10环境下运行yolov3进行目标检测
一、Anaconda 环境配置
首先anaconda环境安装 比较简单,清华镜像安装:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/,
然后next ,最后环境变量设置比较重要,可参考下面链接
https://blog.csdn.net/gyp2448565528/article/details/79451212
为了防止base 环境污染 可以另外 创建环境 conda create -n py3 python=3.5
激活环境 :activate py3
列出所有存在环境 conda env list
删除某环境 conda env remove -n py3
二、tendorflow-gpu 1.9 + CUDA9.0+CUNN
指令 pip install tensorflow-gpu==1.9 -i https://pypi.tuna.tsinghua.edu.cn/simple
CUDA9+CUNN的安装参考 https://www.jianshu.com/p/c228bd790775
安装之后可以添加一下环境变量
三 Keras+yolo3
1. 文件地址 https://github.com/qqwweee/keras-yolo3
2. 原作者采用Darknet 框架实现,在tensorflow需要
将yolov3.weights 转化为可识别的yolov3.h5文件
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
3. python yolo_video.py --image 输入需要检测的图片路径