Darknet主页
Darknet: Open Source Neural Networks in C
安装过程
- clone源码到本地
git clone https://github.com/pjreddie/darknet.git
- 修改Makefile
如果有GPU,并且安装了cuda
GPU=1
如果安装了cudnn
CUDNN=1
如果源码编译了OpenCV
OPENCV=1
不要忘记根据显卡和cuda版本修改ARCH
- 编译安装
make
- 添加环境变量
export PATH="/thirdparty-source/darknet":$PATH
export C_INCLUDE_PATH="/thirdparty-source/darknet/include":$C_INCLUDE_PATH
export CPLUS_INCLUDE_PATH="/thirdparty-source/darknet/include":$CPLUS_INCLUDE_PATH
export LD_LIBRARY_PATH="/thirdparty-source/darknet":$LD_LIBRARY_PATH
export LIBRARY_PATH="/thirdparty-source/darknet":$LIBRARY_PATH
- 测试
darknet imtest /thirdparty-source/darknet/data/eagle.jpg
- 使用预训练的yolov3权重文件检测图片
下载预训练的yolov3权重文件
wget https://pjreddie.com/media/files/yolov3.weights
选择GPU编号0运行
./darknet -i 0 detect cfg/yolov3.cfg /home/zhangchen/yolov3.weights data/dog.jpg
选择CPU运行
./darknet -nogpu detect cfg/yolov3.cfg /home/zhangchen/yolov3.weights data/dog.jpg
注意即使添加了环境变量PATH,这里也必须在darknet文件夹下运行darknet可执行文件,否则会报路径相关的错误
结语
如果您有修改意见或问题,欢迎留言或者通过邮箱和我联系。
手打很辛苦,如果我的文章对您有帮助,转载请注明出处。