基于ubuntu16.04的Yolo V4的keras识别

最新推荐文章于 2020-08-17 13:09:18 发布

未见青山老。

最新推荐文章于 2020-08-17 13:09:18 发布

阅读量493

点赞数

分类专栏：嵌入式系统学习文章标签： python tensorflow ubuntu linux

本文链接：https://blog.csdn.net/qq_41133375/article/details/107146567

版权

8 篇文章 4 订阅

订阅专栏

操作系统：ubuntu16.04
TensorFlow版本：1.14.0
Keras：2.2.5

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

如果 github下载 yolo v4 keras 代码比较慢，可以从 gitee下载：https://gitee.com/yfqh9588/yolov4-keras
在这里插入图片描述

链接：下载地址
提取码：g0ju

1.进入相应文件夹，下载数据集

cd keras-yolo4

wget https://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar

wget https://pjreddie.com/media/files/VOCtest_06-Nov-2007.tar

2.解压数据集

tar xf VOCtrainval_06-Nov-2007.tar
tar xf VOCtest_06-Nov-2007.tar

可以像我这样命名文件夹，后续就不用修改代码。
在这里插入图片描述

可以运行里面的格式转换程序 convert.py。
在这里插入图片描述
在代码文件夹下运行下面代码：

python3 convert.py

在这里插入图片描述
当转换完成之后，便会生成一个h5模型文件。
yolo4_weights.h5可在百度网盘中下载。
链接: 下载链接
提取码: dc2j
yolo4_weights.h5是coco数据集的权重。
yolo4_voc_weights.h5是voc数据集的权重。

gedit test.py

解决python默认路径问题，在文件开头添加如下代码：

import sys
sys.path.remove('/opt/ros/kinetic/lib/python2.7/dist-packages')

在这里插入图片描述
修改使用的模型：

在这里插入图片描述

1.运行代码文件：

cd keras-yolo4
python3 test.py

2.按照提示输入图片名字：
在这里插入图片描述
3.运行结果

关注

专栏目录