Win10下 Tensorflow + YOLOv3的实现
注 意 ! \color{#FF3030}{注意!} 注意!
这里只使用Win10,Ubuntu系统请看原作者的 Github
系统环境,及软件包版本
win10 64位
Python 3.5
Tensorflow 1.11.0
numpy 1.15.1
Pillow 5.3.0
scipy 1.1.0
wget 3.2
seaborn 0.9.0
opencv 4.2.0.32(目前最新版)
easydict 1.9 (目前最新版)
使用Anaconda创建环境
##3 1.新建一个python3.5的环境,名为tensorflow-yolo,系统会自动选择分配一个python3.5.x的版本
conda create --name tensorflow-yolo python=3.5
2.激活新建的环境:activate tensorflow-yolo
左侧的(tensorflow-yolo)就代表进入了新建的虚拟环境
3.安装所需要的各种包
pip install tensorflow==1.11.0
pip install numpy==1.15.1
pip install Pillow==5.3.0
pip install scipy==1.1.0
pip install wget==3.2
pip install seaborn==0.9.0
pip install opencv-python
pip install easydict
创建YOLO环境
1.从Github下载文件,下载完成后解压
2.下载数据集,下载完成后放在tensorflow-yolov3/checkpoint/文件夹下,并解压
3.在tensorflow-yolov3目录下打开cmd窗口,输入
python convert_weight.py
如果出现下面的情况,则成功,此时tensorflow-yolov3/checkpoint/文件夹下应多了3个文件,共计9个文件。
新生成的3个文件:
如果出现下面的情况
ImportError: No module named ‘xxx’
是因为你的环境中缺少这个包,使用 pip install xxx 安装
之后继续输入
python freeze_graph.py
出现下面的清况,则成功
测试
1.测试图片
在tensorflow-yolov3目录下打开cmd窗口,输入
python image_demo.py
2.测试视频
在tensorflow-yolov3目录下打开cmd窗口,输入
python video_demo.py # if use camera, set video_path = 0