- 博客(3)
- 收藏
- 关注
原创 11.13号对pytorch-ssd的尝试
11.13号对pytorch-ssd的尝试 安装各种依赖包 直接pip install 无法安装,采用将包下载下来的方法,对于pytorch采取 pip install +文件后缀名;对于yacs和vizer在github中下载这两个包,在cmd中python setup.py install 的命令 ...
2020-11-03 19:51:36 61
原创 利用c++调用tensorflow中训练的ssd模型
**1.**在tensorflow中训练完成后会生成.ckpt文件,这时需要export_inference_graph.py(路径E:\models-master\research\object_detection)将.ckpt文件转化为.pb文件。 其中只需要修改pipeline_config_path(模型的默认参数文件); trained_checkpoint_prefix(训练产生的.c...
2019-10-31 16:54:03 552
原创 Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize 方法:更新cudnn,我将cudnn更新到7.6 欢迎使用Markdown编辑器 你好! 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Markdown编辑器, 可以仔细阅读这篇文章,了解一...
2019-10-31 14:24:12 298
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人