因参加服创大赛,选题是安全帽检测,所以在这里和大家分享一下简单的用TensorFlow训练yolo v3的过程,适合新手快速上手。
1.下载项目
首先在git clone https://github.com/qqwweee/keras-yolo3将源码clone下来,原来速度太慢的话,可以借助gitee。
在gitee上将项目clone下来:
2.创建虚拟环境
由于作者使用的环境是python3.6+TensorFlow1.6.0+keras2.1.5。所以我们可以先用anaconda创建一个新的虚拟环境,如果创建虚拟环境失败的,可以参照我的这篇博客:conda无法创建虚拟环境。
conda create -n yolo python=3.6 #创建python=3.6的名为yolo的虚拟环境
conda activate yolo #进入新的conda环境
pip install scrapy -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow==1.6.0 #使用清华镜像下载
pip install scrapy -i https://pypi.tuna.tsinghua.edu.cn/simple keras==2.1.5
3.训练前的准备工作
(1)打开pycharm,打开下载的项目
(3)从YOLO网站下载YOLOv3权重
wget https://pjreddie.com/media/files/yolov3.wei