Tensorflow+SSD+Windows10+GPU的配置及运行

来源:互联网 发布:网络电话机 编辑:程序博客网 时间:2024/05/16 08:17

最近在电脑的上配置运行了SSD算法,记录下流程(亲测有效)
1、从https://github.com/balancap/SSD-Tensorflow 上下载源码并解压
2、下载ssd_300_vgg.ckpt.zip,解压放在SSD-Tensorflow-master/checkpoints文件夹中
这里需要注意的是,可能会出现路径不对的错误,建议使用绝对路径,如果绝对路径还是不行的话,将模型和运行文件放在同一路径下尝试。
3、运行ssd_notebook.ipynb文件。
需要用到插件软件jupyter-notebook:在终端进入SSD-Tensorflow-master下的notebook目录,运行jupyter ssd_notebook.ipynb
或者如果没有这个软件,也可以自己建一个.py文件,将里面的代码复制粘贴过去,运行即可
4、改变path = ‘../demo/’成自己的图片所在路径(建议使用绝对路径)
5、从头到尾运行所有的cell(如果你使用的是jupyter)

原创粉丝点击