darknet安装
git clone https://github.com/pjreddie/darknet
cd darknet
make
这里由于我需要在服务器,gpu运行,先对Makefile文件进行一下修改
GPU=1
CUDNN=1
OPENCV=1
注意千万不要忘了修改nvcc
nvcc=/usr/local/cuda-8.0/bin/nvcc
如果忘了改可能会出现如下错误:/bin/sh: 1: nvcc: not found
改完之后:
下面复制这个darknet到服务器上。
scp -r /home/papa/Downloads/darknet ywshang@172.18.32.238:/home/ywshang/
这条命令的作用是复制指定文件夹到服务器上。登录服务器make一下吧!
之前做这件事还出过一个错误
libcudart.so.8.0: cannot open shared object file: No such file or directory
服务器root权限下运行这句话就ok了~
sudo ldconfig /usr/local/cuda/lib64
详情请看我的这篇博客:
http://blog.youkuaiyun.com/shangpapa3/article/details/76189674
成功现在让我们来测试一下~
wget https://pjreddie.com/media/files/yolo.weights
./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg
或
./darknet detector test cfg/coco.data cfg/yolo.cfg yolo.weights data/dog.jpg
运行成功
这里根据官网步骤简单运行了一下,更多功能设置,请看官网。
https://pjreddie.com/darknet/yolo/