YOLO3搭建数据集前的准备
yolo3下载地址:https://github.com/qqwweee/keras-yolo3
-
使用的工具 pycharm
本人显卡:英伟达 RTX2070
tensorflow-gpu安装前要安装显卡驱动:CUDA CUDNN 版本安装的是10.0
tensorflow官网 点击查看版本信息
python 3.6.5 Anaconda(第三方用来安装python库,建议安装Anaconda3 4.2.0版本)
tensorflow—gpu 1.13.1
opencv 3.4.1.15
keras 2.2.4 (注意,tensorflow的版本和keras版本要兼容,版本不一样会出问题会出问题。)
安装keras时注意一个问题那就是用库之间版本的兼容问题。
cuda下载地址:https://developer.nvidia.com/cuda-downloads
1更换源
anaconda更换清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls yes
pip更换国内源
(1)临时使用:
可以在使用pip的时候,加上参数-i和镜像地址(如
https://pypi.tuna.tsinghua.edu.cn/simple),
例如:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy
(2)永久修改
Linux:修改 ~/.pip/pip.conf (没有就创建一个文件夹及文件。文件夹要加“.”,表示是隐藏文件夹)
内容如下:
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host = https://pypi.tuna.tsinghua.edu.cn
windows:直接在user目录中创建一个pip目录,如:C:\Users\xx\pip,然后新建文件pip.ini,即 %HOMEPATH%\pip\pip.ini,在pip.ini文件中输入以下内容(以豆瓣镜像为例):
[global]
index-url = http://pypi.douban.com/simple
[install]
trusted-host = pypi.douban.com
1下载安装keras 2.2.4
keras的中文文档:https://keras-cn.readthedocs.io
keras是基于Tensorflow,Theano以及CNTK后端时的依赖包,也就是说要想安装keras必须至少选择安装tensorflow,安装Theano和CNTK一种,我用的是Tensorflow
#我用的是anaconda 里的prompt
pip install keras==2.2.4
第一步制作voc数据集
voc2007数据集 链接:https://pan.baidu.com/s/1T8k7vQG1eXtuMcaVcgwvCQ 提取码:l7vz
1 打开后保持目录结构,删除所有文件夹里面的文件。
注意保留convert_to_txt.py文件
2 利用打标签工具给图片打标签。labelimg.exe
链接:https://pan.baidu.com/s/1g18REDGtFDOBmP-D68b01g 提取码:7kmt
注意在使用时要把 data和labelimg.exe 拖拽到 桌面,这样才能正常使用
1 打开labelimg.exe
快捷键:
d 下一张图片
a上一张图片
w编辑图框
图片的获取办法参考我的另一篇博客,通过视频拆分帧获取用来打标签的图片