Tensorflow-yolov3代码复现

版权声明:本文为博主原创文章,未经博主允许不得转载。如有问题,欢迎指正。 https://blog.csdn.net/kellyroslyn/article/details/88977588

此博文仅供个人学习之用!

1.复制这个文件:

$ git clone https://github.com/YunYang1994/tensorflow-yolov3.git

2.在食用代码前,先安装一些依赖项:

$ cd tensorflow-yolov3
$ pip install -r ./docs/requirements.txt

步骤截图如下:
在这里插入图片描述
在这里插入图片描述
3.把加载好的COCO权重导出为TF checkpoint (yolov3.ckpt) 和 frozen graph (yolov3_gpu_nms.pb) 。
如果没有yolov3.weights的话,去下载,然后放到./checkpoint目录下。下载地址是:
https://github.com/YunYang1994/tensorflow-yolov3/releases/download/v1.0/yolov3.weights

$ python convert_weight.py --convert --freeze

步骤截图如下:
在这里插入图片描述
在这里插入图片描述
4.然后,./checkpoint目录下就会出现一些.pb文件。现在可以跑Demo脚本了:

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

步骤截图如下:
在这里插入图片描述
跑nms_demo.py的时候出现:

TypeError: rectangle() got an unexpected keyword argument 'width'

的错误。
解决方法是:将tensorflow-yolov3/core/utils.py中的180行代码去掉 ‘,width=3’,重新运行代码即可。
在这里插入图片描述
跑python video_demo.py,因为没有调用camera,所以将video_demo.py第24行代码注释掉,运行代码即可。

猜你喜欢

转载自blog.csdn.net/kellyroslyn/article/details/88977588
今日推荐