Ubuntu下darknet yolo4的编译
Ubuntu下darknet yolo4的编译
首次编辑：24/5/17/23:16
最后编辑：24/5/18/12:47
参考链接
- YOLOv4-darknet installation and usage on your system (Windows & Linux)
- Installing and Building Darknet
- 官网
0 说明
此博客只记录darknet yolo4 + opencv的编译安装，不涉及cuda和cuDNN。
1 编译的源文件
darknet的github地址为https://github.com/AlexeyAB/darknet。
首先要明确darknet这个darknet的版本，建议点开仓库中的
cfg文件夹，看看里面有没有以yolov4开头的文件，有的话才是yolo4版本。
之前第一次编译的时候很神奇地编译了yolo3版本，但明明github链接地址看起来是一样的，还是谨慎为妙。
1.1 克隆仓库
git clone https://github.com/AlexeyAB/darknet.git
建议后面这个仓库链接还是直接去GitHub上复制。
1.2 修改Makefile
克隆完仓库之后，进入darknet文件夹。里面有个叫Makefile的文件，用代码/文本编辑器打开它。
在Makefile文件的前几行中：
GPU=0 # gpu加速，应该就是配合cuda toolkit的
CUDNN=0 # 这个应该也是cuda加速的
CUDNN_HALF=0 # 同上
OPENCV=0
AVX=0 # x86架构处理器指令集的一个扩展，用于提高并行计算的速度
OPENMP=0 # Open Multi-Processing，cpu加速
LIBSO=0 # 是否将darknet编译成静态库
ZED_CAMERA=0 # to enable ZED SDK 3.0 and above
ZED_CAMERA_v2_8=0 # to enable ZED SDK 2.X
如果没有这么多选项那么应该是克隆错版本了。
将OPENCV = 0改为OPENCV = 1，如果有cuda和cudnn的话就把前三行的值都改为1。其它的几个选项可以根据自己的电脑情况和需求自行更改。
一般操作到这里就改完了，但由于我的opencv是通过源码编译安装的，还需要改一点东西。
如果opencv是源码编译的
在Makefile文件中找到下面片段：
ifeq ($(OPENCV), 1)
COMMON+= -DOPENCV
CFLAGS+= -DOPENCV
LDFLAGS+= `pkg-config --libs opencv4 2> /dev/null || pkg-config --libs opencv`
COMMON+= `pkg-config --cflags opencv4 2> /dev/null || pkg-config --cflags opencv`
endif
将LDFLAGS+=后面的值改为
-L/path/to/opencv/lib -lopencv_gapi -lopencv_highgui -lopencv_ml -lopencv_objdetect -lopencv_photo -lopencv_stitching -lopencv_video -lopencv_calib3d -lopencv_features2d -lopencv_dnn -lopencv_flann -lopencv_videoio -lopencv_imgcodecs -lopencv_imgproc -lopencv_core
其中path/to/opencv/是opencv的路径。
然后将COMMON+=后面的值改为
-I/path/to/opencv/include/opencv4
1.3 执行Make
打开终端，进入darknet文件夹，执行make。等待编译完成。
2 测试darknet
2.1 测试可执行文件
打开终端，在darknet文件夹下，执行./darknet，若输出usage: ./darknet <function>，则表示编译成功。
2.2 测试物体识别功能
首先为了测试，我们要先下载一下别人训练好的权重文件。打开终端，在darknet文件夹下，执行：
wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights
检测图片
接着检测一下cv小狗的图片
./darknet detector test cfg/coco.data cfg/yolov4.cfg yolov4.weights data/dog.jpg
正常情况下，将会弹出一个窗口，正是那张经典的cv小狗加自行车的检测图像。
检测视频
没有装cuda的情况下，检测视频可能会非常慢，展示检测结果的窗口可能会很久才弹出。
./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights <video file>
把<video file>替换成你的视频文件名。
或者也可以将检测的结果保存成一个视频：
./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights <video file> -out_filename <output_video file>
电脑相机实时检测
没有装cuda的情况下，实时检测可能会非常慢，展示检测结果的窗口可能会很久才弹出，甚至根本就没法弹出这个窗口，只会显示进程失去响应。
./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights
