- 选择安装目录,进入
git clone https://github.com/AlexeyAB/darknet.git
- 源码文件夹下的README.md文件介绍了安装方式,和其它需要了解的东西
- 根据系统实际情况进行配置,如果需要使用GPU进行加速,则在Makefile文件中修改前四行
GPU=1 CUDNN=1 CUDNN_HALF=1 OPENCV=1
我在安装过程中全部没有开启,如有需要开启opencv,后面再安装,先确定yolov4可以运行 - 随后使用make或者是make -j8重新编译。生成darknet文件,直接使用./darknet 显示如下结果:
usage: ./darknet - 下载之后将yolov4.weights放在YOLOv4的主目录下即可进行测试,命令如下:
./darknet detect cfg/yolov4.cfg yolov4.weights data/dog.jpg
运行成功后会生成一张图片,以及图像预测结果
在树莓派上会因为树莓派内存小而out of memory,这时需要改动cfg/yolov4.cfg文件中的图片分辨率,height和width,原来是416x416或者是608x608,改小测试出能用的即可,subdivisions参数是训练时用到的参数,表示将图片分为多少次送入网络,这样会降低对内存的占用。