.
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
(作者人巨好)