![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
hustwayne
。。。。。。
展开
-
tensorflow 加载多个pb模型节点冲突解决
使用多个pb模型时,有时会遇到模型间节点相同的情况,造成图加载冲突,无法输出正确结果,更改节点信息较为麻烦,因此创建新图来解决 下面是直接使用默认图tf.get_default_graph(), 如果模型间无冲突可使用。 def __init__(self, graph_path, target_size=(300, 300), confidence_thresh=0.9): self.target_size = target_size ...原创 2020-12-01 14:51:45 · 715 阅读 · 0 评论 -
Tensorflow 错误集
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try loo。。。。。 解决方法:指定gpu import os import os os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0' ...原创 2019-11-28 15:59:28 · 167 阅读 · 0 评论