Darknet YoloV4编译+训练(避免踩坑)

AlexAB/darknet-yolov4编译+训练

编译

官方源码地址奉上:

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

原地解压

  • unzip darknet-master.zip ///如使用上述git clone 下载源码可跳过解压
  • cd darknet

你会看到一系列文件:
Darknet YoloV4编译+训练(避免踩坑)_第1张图片找到Makefile

  • gedit Makefile or 使用图形界面 对该文件进行修改
 - GPU=1
   CUDNN=1
   CUDNN_HALF=1
   OPENCV=1
   AVX=0
   OPENMP=0
   LIBSO=1
   ZED_CAMERA=0
   ZED_CAMERA_v2_8=0
=====================
	USE_CPP=0
	DEBUG=0
  • 这个很重要(关系到你后期编译成功与否)我所用的机器算力为7.5,所以要在ARCH= 中加入该行
    Darknet YoloV4编译+训练(避免踩坑)_第2张图片
  • makefile文件中第77行左右:

NVCC = nvcc(此处需要设置你机器的绝对路径)

  • 至此Ctrl + s(保存文件)

编译

  • 首先你需要执行:

make clean

去初始化一下该项目

  • 开始编译:

make or make -j8
(接下来就是编译过程了,make就行,如果想多线程下载加快下载速度可以用make -j8,不过还是注意电脑性能,线程太多Linux可能会死机)

  • 如果过程中出现
> Makefile:185: recipe for target ‘obj/network_kernels.o‘ failed make: ***
> 
  • 去工程中的src文件夹中找到network kernels cu
    这个文件721行注释后 Ctrl+s保存
    Darknet YoloV4编译+训练(避免踩坑)_第3张图片
  • 再次执行

make clean

make or make -j8

  • 终端输入 :

./darknet
:usage: ./darknet

  • 至此编译成功!

你可能感兴趣的:(目标检测yolov4,计算机视觉,opencv,人工智能)