配置环境
visual studio 2013
显卡 GTX 960M
CUDA 7.5
OpenCV 2.4.9
pthread 下载地址
YOLO官网
[http://pjreddie.com/darknet/yolo/]
步骤
CUDA 版本安装教程:
CPU版本安装步骤:
第一步:建立vs2013 x86(x64)位项目程序
第二步:按照Makefile文件,将YOLO官网中下载的以下 * .c ,*.h 文件添加到工程
gemm.o utils.o cuda.o deconvolutional_layer.o convolutional_layer.o list.o image.o activations.o im2col.o col2im.o blas.o crop_layer.o dropout_layer.o maxpool_layer.o softmax_layer.o data.o matrix.o network.o connected_layer.o cost_layer.o parser.o option_list.o darknet.o detection_layer.o imagenet.o captcha.o route_layer.o writing.o box.o nightmare.o normalization_layer.o avgpool_layer.o coco.o dice.o yolo.o layer.o compare.o classifier.o local_layer.o swag.o shortcut_layer.o
第三步:配置opencv, pthread 库
opencv 包含路径
opencv 库路径
pthread 包含路径
pthread 库路径
第四步:修改错误
a、将unistd.h去掉,此头文件只在Linux c存在
b、将程序中关键字 class 改名,不然会和c++代码的类名冲突
c、将gcc中随机函数改成 rand()
d、还有其他的错一大堆….需要细心的修改
第五步:运行
运行之前需要下载深度学习网络训练模型,还好作者提供了3个版本的model:
下载地址:这里写链接内容
程序运行命令行: yolo test cfg/yolo-tiny.cfg yolo-tiny_1000.weights
经过以上步骤折腾,基本都可以运行程序了,现在来看看程序运行效果:
input image
out result
完整github工程代码链接地址:
https://github.com/guozhongluo/YOLO
联系方式QQ: 472338311
邮箱: [email protected]