首页 > 科技 >

简洁几步实现Keras下使用YOLO V3方针检测

2019-05-29 06:14:47 暂无 阅读:1989 评论:0

YOLO V3的介绍好多,本文不再胪陈,仅列举在keras下完成方针检测需要的几个步伐。

1、从GitHub下载项目https://github.com/qqwweee/keras-yolo3,解压缩到yolo3目录;

2、下载 YOLOv3 权重wget https://pjreddie.com/media/files/yolov3.weights,并拷贝到yolo3目录中;

3、转换为keras模型,打开终端,进入yolo3目录,输入python3 convert.py yolov3.cfg yolov3.weights model_data/yolo.h5

4、进行图像检测from yolo import YOLO

from PIL import Image

yolo = YOLO()

image = Image.open("th.jpg")

r_image = yolo.detect_image(image)

r_image.show()

在yolo3目录中存放了一张名为th的企鹅图片,检测究竟如下

简洁几步实现Keras下使用YOLO V3方针检测

Keras下经由YOLO V3检测企鹅图片的究竟

同时有检测究竟输出,花样为:类别 概率 (中心坐标X,中心坐标Y) (界限框宽,界限框高)

Found 2 boxes for img

bird 0.62 (89, 65) (249, 312)

bird 1.00 (279, 80) (388, 322)

若是需要摄像头实时识别,可经由opencv挪用。

相关文章