Tensorflow-yolov3代码复现

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

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行代码注释掉，运行代码即可。