2020-10-14

在运行https://github.com/MonkeyDTH/h52pb中的load_pb_test.py时出现以下的错误:

File "load_pb_test.py", line 67, in <module>
    recognize(img, pb_path)
  File "load_pb_test.py", line 16, in recognize
    tensors = tf.import_graph_def(output_graph_def, name="")
  File "/home/ed/anaconda3/envs/cam/lib/python3.6/site-packages/tensorflow/python/framework/importer.py", line 313, in import_graph_def
    op_def=op_def)
  File "/home/ed/anaconda3/envs/cam/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 2630, in create_op
    original_op=self._default_original_op, op_def=op_def)
  File "/home/ed/anaconda3/envs/cam/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1204, in __init__
    self._traceback = self._graph._extract_stack()  # pylint: disable=protected-access

InvalidArgumentError (see above for traceback): NodeDef mentions attr 'explicit_paddings' not in Op<name=Conv2D; signature=input:T, filter:T -> output:T; attr=T:type,allowed=[DT_HALF, DT_FLOAT]; attr=strides:list(int); attr=use_cudnn_on_gpu:bool,default=true; attr=padding:string,allowed=["SAME", "VALID"]; attr=data_format:string,default="NHWC",allowed=["NHWC", "NCHW"]>; NodeDef: conv2d/Conv2D = Conv2D[T=DT_FLOAT, data_format="NHWC", dilations=[1, 1, 1, 1], explicit_paddings=[], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/gpu:0"](_arg_conv2d_input_0_0/_1, conv2d/Conv2D/ReadVariableOp). (Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary.).
         [[Node: conv2d/Conv2D = Conv2D[T=DT_FLOAT, data_format="NHWC", dilations=[1, 1, 1, 1], explicit_paddings=[], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/gpu:0"](_arg_conv2d_input_0_0/_1, conv2d/Conv2D/ReadVariableOp)]]

当前环境:conda 虚拟环境中

tensorflow                1.3.0                         0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
tensorflow-base           1.3.0            py36h5293eaa_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
tensorflow-gpu            1.3.0                         0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
tensorflow-gpu-base       1.3.0           py36cuda8.0cudnn6.0_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
tensorflow-tensorboard    0.1.8                    pypi_0    pypi
theano                    0.9.0                    py36_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
tk                        8.6.10               hbc83047_0    defaults
toolz                     0.8.2                    py36_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
 

于是更换环境,将tensorflow-gpu换成1.14的,再次输入命令python load_pb_test.py,运行成功。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值