刚刚安装测试了YOLOv5版本,本文继续安装测试一下YOLOv4版本,并在后续做一些对比。闲言少叙,首先下载源代码文件:
git clone https://github.com/AlexeyAB/darknet.git
如果没有git命令行,除了安装之后重新运行上一命令之外,也可以直接去github上下载源代码包。
随后根据系统实际情况进行配置,如果需要使用GPU进行加速,则在Makefile文件中修改前四行
GPU=1
CUDNN=1
CUDNN_HALF=1
OPENCV=1
另外 DEBUG也可以设置为1
随后使用make或者是make -j8重新编译。生成darknet文件,直接使用./darknet 显示如下结果:
DEBUG=1
usage: ./darknet <function>
然后可以下载预训练的权重,具体网址:https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights
下载之后将yolov4.weights放在YOLOv4的主目录下即可进行测试,命令如下:
./darknet detect cfg/yolov4.cfg yolov4.weights data/dog.jpg
实际显示效果:
------------------------------ 正文到此结束---------------------------------
推荐一个公众号:健哥聊量化,会持续推出股票相关基础知识,以及python实现的一些基本的分析代码。欢迎大家关注,二维码如下:
相关文章列表如下: