Question

我从中下载了一个预先训练好的resnet pb文件 http://download.tensorflow.org/models/official/resnet_v1_imagenet_savedmodel.tar.gz

当我尝试通过以下代码将其导入图表时

tf.Graph().as_default():

    with tf.gfile.FastGFile(path, 'rb') as file:
        graph_def = tf.GraphDef()

        graph_def.ParseFromString(file.read())

        tf.import_graph_def(graph_def, name='')`

我收到了错误

    graph_def.ParseFromString(file.read())
  File "D:\Anaconda3\lib\site-packages\google\protobuf\message.py", line 185, in
 ParseFromString
    self.MergeFromString(serialized)
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\python_message.p
y", line 1083, in MergeFromString
    if self._InternalParse(serialized, 0, length) != length:
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\python_message.p
y", line 1120, in InternalParse
    pos = field_decoder(buffer, new_pos, end, self, field_dict)
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\decoder.py", lin
e 633, in DecodeField
    if value._InternalParse(buffer, pos, new_pos) != new_pos:
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\python_message.p
y", line 1120, in InternalParse
    pos = field_decoder(buffer, new_pos, end, self, field_dict)
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\decoder.py", lin
e 612, in DecodeRepeatedField
    if value.add()._InternalParse(buffer, pos, new_pos) != new_pos:
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\python_message.p
y", line 1120, in InternalParse
    pos = field_decoder(buffer, new_pos, end, self, field_dict)
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\decoder.py", lin
e 743, in DecodeMap
    if submsg._InternalParse(buffer, pos, new_pos) != new_pos:
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\python_message.p
y", line 1109, in InternalParse
    new_pos = local_SkipField(buffer, new_pos, end, tag_bytes)
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\decoder.py", lin
e 850, in SkipField
    return WIRETYPE_TO_SKIPPER[wire_type](buffer, pos, end)
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\decoder.py", lin
e 799, in _SkipGroup
    new_pos = SkipField(buffer, pos, end, tag_bytes)
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\decoder.py", lin
e 850, in SkipField
    return WIRETYPE_TO_SKIPPER[wire_type](buffer, pos, end)
  File "D:\Anaconda3\lib\site-packages\google\protobuf\internal\decoder.py", lin
e 814, in _SkipFixed32
    raise _DecodeError('Truncated message.')
google.protobuf.message.DecodeError: Truncated message.

＆＃39;路径＆＃39;是pb文件。

Answer 1

我发现来自http://download.tensorflow.org/models/official/resnet_v1_imagenet_savedmodel.tar.gz的pb文件只有806KB。所以它可能是没有权重的图形文件

无法从Tensorflow的官方网站

1 个答案: