yolov2在CUDA8.0+cudnn8.0下安装、训练、检测经历

这次用yolov2做检测时遇到个大坑,折腾了我好几天,特以此文记录之。

一、安装cuda+cudnn

      它们的版本必须要匹配,否则训练后检测不出目标!

      1、下载cuda8.0.61_375.26_linux.run和cudnn-8.0-linux-x64-v5.1.tgz

      2、安装cuda

           sh cuda8.0.61_375.26_linux.run

           一直enter,直至accept。然后是安装一些东西直接y,或者让你输入地址,

 按enter用默认的就可以了。

      3、环境变量设置

       sudo gedit ~/.bashrc

       在其末尾添加: 

     export CUDA_HOME=/usr/local/cuda-8.0

 

    export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

 

    export PATH=/usr/local/cuda-8.0/bin:$PATH

       
   保存,刷新:source ~/.bashrc

    4、动态链接设置

        创建文件:sudo gedit /etc/ld.so.conf.d/cuda.conf

        写入:/usr/local/cuda/lib64

        保存:sudo ldconfig -v

   装好后,可用nvcc --version查看cuda版本

  5、安装cudnn

      如果之前安装了cudnn,可删除之: 

    cd usr/local/cuda/lib64
    sudo rm libcudnn*
解压:
tar -zxvf cudnn-8.0-linux-x64-v5.1.tgz
cd ../lib64
sudo cp lib* /usr/local/cuda/lib64/
cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include/
更新软连接:
cd /usr/local/cuda/lib64/

sudo rm -rf libcudnn.so libcudnn.so.5

sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5

sudo ln -s libcudnn.so.5 libcudnn.so

更新:sudo ldconfig

 参考:http://blog.csdn.net/jhszh418762259/article/details/52958287?locationNum=8&fps=1

            http://www.cnblogs.com/denny402/p/5067265.html

          http://www.jianshu.com/p/3c06b2a6b9df

二、yolov2安装测试

yolov2官网:https://pjreddie.com/darknet/yolo/

官网上有些安装步骤说的不是很清楚,所以很容易遇到坑。

1、darknet安装:git clone https://github.com/pjreddie/darknet.git

不要忘了修改makefile和nvcc!

GPU,cudnn,opencv设为1

nvcc=/usr/local/cuda-8.0/bin/nvcc

然后用make指令编译

2、下载voc数据集

    用作者给的指令下载没有成功,手动下载即可。

     在scripts里面转换格式:python voc_label.py

     生成train文件:cat 2007_* 2012_train.txt > train.txt

     在cfg/voc.data中修改训练数据路径

3、下载预训练模型

     作者提供了很多其训练好的模型

4、自己训练voc数据

     ./darknet detector train ./cfg/voc.data ./cfg/yolo-voc.cfg

5、测试下迭代效果,比如6000次

   ./darknet detector test cfg/voc.data cfg/yolo-voc.cfg backup/yolo-voc_6000.weights data/dog.jpg

 

参考:http://blog.csdn.net/samylee/article/details/53414108

        http://blog.csdn.net/u014540717/article/details/53411948

 

 

 这个论坛中,作者会亲自解答yolov2:https://groups.google.com/forum/#!forum/darknet

     

 

转载于:https://www.cnblogs.com/573177885qq/p/6632576.html

你可能感兴趣的:(git,python,人工智能)