tensorflow yolo v3代码实现

.

tensorflow yolo v3

参考：https://github.com/YunYang1994/tensorflow-yolov3
（作者人巨好）

环境xshell gpu服务器

1

git源码

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

2

权重文件拷贝及下载

$ cd tensorflow-yolov3
$ pip install -r ./docs/requirements.txt
$ cd checkpoint
$ wget https://github.com/YunYang1994/tensorflow-yolov3/releases/download/v1.0/yolov3_coco.tar.gz
$ tar -xvf yolov3_coco.tar.gz
$ cd ..
$ python convert_weight.py
$ python freeze_graph.py

解压完的模型

3 溢出问题

看到类似
InternalError: Dst tensor is not initialized.
考虑溢出问题

出现这个错误一般是GPU内存耗尽，挂在后台进程中的IPython内核会这样做

运行程序之前，先运行export CUDA_VISIBLE_DEVICES=1，仅显卡设备1GPU可见，tensorflow 算是一个比较贪心的工具了就算用device_id指定gpu 也会占用别的GPU的显存资源 必须在执行程序前执行
export CUDA_VISIBLE_DEVICES=n（n为可见的服务器编号）再去执行Python 代码.py 才不会占用别的GPU资源

原文：https://blog.csdn.net/wanglitao588/article/details/77033659

3 初步运行截图

再次申明：参考：https://github.com/YunYang1994/tensorflow-yolov3
（作者人巨好）