这些个事儿

使用多个pb模型时，有时会遇到模型间节点相同的情况，造成图加载冲突，无法输出正确结果，更改节点信息较为麻烦，因此创建新图来解决 下面是直接使用默认图tf.get_default_graph()， 如果模型间无冲突可使用。 def __init__(self, graph_path, target_size=(300, 300), confidence_thresh=0.9): self.target_size = target_size ...

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' ...