You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.6 KiB
61 lines
1.6 KiB
9 years ago
|
# Copyright (c) 2016 Baidu, Inc. All Rights Reserved
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
"""
|
||
|
|
Show the content of proto buffer data file of PADDLE
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from google.protobuf.internal.decoder import _DecodeVarint
|
||
|
|
import paddle.proto.DataFormat_pb2 as DataFormat
|
||
|
|
|
||
|
|
def read_proto(file, message):
|
||
|
|
"""
|
||
|
|
read a protobuffer struct from file, the length of the struct is stored as
|
||
|
|
a varint, then followed by the actual struct data.
|
||
|
|
@return True success, False for end of file
|
||
|
|
"""
|
||
|
|
|
||
|
|
buf = file.read(8)
|
||
|
|
if not buf:
|
||
|
|
return False
|
||
|
|
result, pos = _DecodeVarint(buf, 0)
|
||
|
|
buf = buf[pos:] + file.read(result - len(buf) + pos)
|
||
|
|
message.ParseFromString(buf)
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def usage():
|
||
|
|
print >>sys.stderr, "Usage: python show_pb.py PROTO_DATA_FILE"
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
if len(sys.argv) < 2:
|
||
|
|
usage()
|
||
|
|
|
||
|
|
f = open(sys.argv[1])
|
||
|
|
header = DataFormat.DataHeader()
|
||
|
|
read_proto(f, header)
|
||
|
|
print header
|
||
|
|
|
||
|
|
sample = DataFormat.DataSample()
|
||
|
|
while read_proto(f, sample):
|
||
|
|
print sample
|
||
|
|
|
||
|
|
|
||
|
|
|