Yolo v4 keras识别(Ubuntu18)

Yolo v4 keras识别(Ubuntu18)

一、darknet测试

下载darknet:

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

Yolo v4 keras识别(Ubuntu18)_第1张图片
进入到darknet目录下进行编译:
Yolo v4 keras识别(Ubuntu18)_第2张图片
测试:若出现以下,说明编译成功。
在这里插入图片描述
生成上述视频的命令:

./darknet.exe detector demo ../../cfg/coco.data ../../cfg/yolov4-1024.cfg ../../yolov4.weights ~/Desktop/0002-20170519-2.mp4 -thresh 0.2 -ext_output -out_filename ~/Desktop/output.avi

二、keras识别

下载keras和tensorflow:
这两者的版本要匹配,参考地址

比如Python2.7:

sudo pip3 install keras==2.2.5
sudo pip3 install tensorflow==1.14.

下载Keras源码:

git clone https://github.com/Ma-Dan/keras-yolo4

Yolo v4 keras识别(Ubuntu18)_第3张图片
下载权重文件并保存在keras源码包中:
链接:https://pan.baidu.com/s/1FF79PmRc8BzZk8M_ARdMmw
提取码:dc2j
Yolo v4 keras识别(Ubuntu18)_第4张图片
导入一张图片到keras代码包中,
把这里改为自己的图片名:
Yolo v4 keras识别(Ubuntu18)_第5张图片
使用yolov4_weight.h5:
修改test.py:
Yolo v4 keras识别(Ubuntu18)_第6张图片
运行test.py:

python test.py

Yolo v4 keras识别(Ubuntu18)_第7张图片Yolo v4 keras识别(Ubuntu18)_第8张图片Yolo v4 keras识别(Ubuntu18)_第9张图片
使用yolov4_voc_weight.h5:
修改test.py:
Yolo v4 keras识别(Ubuntu18)_第10张图片
运行:

python test.py

Yolo v4 keras识别(Ubuntu18)_第11张图片Yolo v4 keras识别(Ubuntu18)_第12张图片

参考文献:
https://blog.csdn.net/qq_42451251/article/details/107137508

你可能感兴趣的:(python,tensorflow,ubuntu)