TensorFlow2.x-YOLOv3在Win10中运行Demo

TensorFlow2.x-YOLOv3是一个基于TensorFlow2.X版本实现的YOLO3仓库,支持训练和测试自己的数据库。要训练自己的数据库,首先要确认该仓库在当前环境下可以成功运行其demo,本文记录在Win10+Python3.6环境下成功运行demo的过程。

致谢

tensorflow-yolov3 基于tensorflow-gpu==1.11.0版本实现的yolo3版本

TensorFlow2.0-Examples基于tensorflow-gpu==2.0版本实现的yolo3版本

向原作者致敬,非常感谢!

运行环境

Win10 + Python 3.6

TensorFlow-cpu 2.2

Keras 2.3.1

TensorFlow2.x-YOLOv3运行Demo

1. 下载TensorFlow2.x-YOLOv3

2. 安装必要项

在运行此命令之前,我们先打开TensorFlow2.0-Examples/4-Object_Detection/YOLOV3/docs/requirements.txt文件看一下:

如上所示,tensorflow的版本要求为2.0.0,且默认会安装GPU版本。

如果当前电脑中没有安装过tensorflow任何版本,且支持GPU,可以直接运行下面的安装命令。

如果电脑不支持GPU且未安装过任何tensorflow版本,请将tensorflow==2.0.0修改为tensorflow-cpu==2.2,再运行下面的安装命令。或者可以删掉tensorflow==2.0.0,先单独安装tensorflow

如果电脑中已有tensorflow2.0以上任意版本,可以删除tensorflow==2.0.0,再运行下面的安装命令。

这里需要特别注意,使用pip安装tensorflow时,并不会先卸载再安装,会让电脑中同时存在多个版本。

我在安装必要项时,踩过的坑,供大家参考:

在此之前,我已安装TensorFlow-cpu 2.2版本。然后我先尝试安装tensorflow-yolov3,于是我电脑中便同时有了TensorFlow-cpu 2.2和tensorflow-gpu==1.11.0版本,导致后续错误百出。最后我卸载了所有的tensorflow版本,重新安装TensorFlow-cpu 2.2,将requirements.txt中的tensorflow==2.0.0删除,再运行此步的安装命令。

上述命令运行完成后,请确认当前电脑中是否已安装keras。

无论是运行TensorFlow2.x-YOLOv3的demo,还是后续训练自己的数据库,都需要keras,但是requirements.txt没有写,所以请参考这里自行安装。

3. 下载yolov3.weights

打开cmd.exe,来到TensorFlow2.0-Examples/4-Object_Detection/YOLOV3/路径下,下载yolov3.weights。

4. 运行image_demo

打开cmd.exe,来到TensorFlow2.0-Examples/4-Object_Detection/YOLOV3/路径下,运行下面的命令:

tensorflow2.x-yolov3

下一篇文章将分享如何使用TensorFlow2.x-YOLOv3训练自己的数据库。

本文到此结束,感谢阅读,谢谢支持。

发表评论

电子邮件地址不会被公开。 必填项已用*标注

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据

Fork me on GitHub