试玩YOLOv3多目标检测

YOLOv3简介

官网链接
论文链接

运行环境搭建

首先下载源码:
源码: https://github.com/NUISTGY/TensorFlow2.0-Examples/tree/master/4-Object_Detection/YOLOV3

接着安装依赖库和训练好的权重:

1
2
$ pip3 install -r ./docs/requirements.txt
$ wget https://pjreddie.com/media/files/yolov3.weights

将权重文件置于源码文件夹目录下即可。

Quick Start

前面已经完成了所有的准备公作,运行以下命令即可进行目标检测:

1
2
$ python image_demo.py
$ python video_demo.py # if use camera, set video_path = 0

运行python image_demo.py的结果

尝试检测自己的图片或视频

想要识别自己的图片只需要做很小的改动即可。
打开image_demo.py:

将22行image_path改为自己图片的地址即可,视频目标检测同理。

最后再贴一张目标比较多的照片:

文章作者: GeYu
文章链接: https://nuistgy.github.io/2019/10/24/试玩YOLOv3多目标检测/
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 Yu's Blog