利用自己的数据进行AB-darknet yolov3的训练

要求

• Linux or Windows
• CUDA >= 10.0
• Opencv >=2.4
• cndnn >=7.0 for CUDA
• GPU with CC>=3.0
• Linux: GCC or Clang

下载

下载代码的压缩包或者执行如下指令：

• git clone https://github.com/AlexeyAB/darknet.git
• cd darknet

进入darknet的根目录下

MAKE

首先修改Makefile， 如果你要用到GPU，那么第一行设置为

• GPU=1

如果要利用cudnn加速，那么第二行设置为

• CUDNN=1

如果要和opencv一起编译，那么第四行设置为

• OPENCV=1

之后输入指令

• make

在这个过程中有可能会碰到错误：

YOLOv3darknet:/make:[obj/convolutional_kernels.o] Error 127

这时候是因为nvcc的环境变量设置有问题，我的方法是修改Makefile,修改nvcc那一行为：

• NVCC=/usr/local/cuda/bin/nvcc

即可，之后再输入make，等待一段时间后就可以顺利编译完成。