python hdf5

import h5py
import numpy as np


# the train.h5 is the annotation of pascal3d

f = h5py.File('train.h5', 'r')
print f.keys()
a_center = f['center'][:]
print a_center
file_center = open("test_center.txt", 'w')
print type(a_center)
print len(a_center)
print a_center[1][1]
for i in range (len(a_center)):
    file_center.write(str(a_center[i][0]))
    file_center.write(" ")
    file_center.write(str(a_center[i][1]) + '\n')

file_center.close()


a_imgname = f['imgname'][:]
print a_imgname
file_imgname = open("test_imgname.txt", 'w')
print type(a_imgname)
print len(a_imgname)

for i in range (len(a_imgname)):
    file_imgname.write(str(a_imgname[i]) + '\n')

file_imgname.close()


a_index = f['index'][:]
print a_index
file_index = open("test_index.txt", 'w')
print type(a_index)
print len(a_index)

for i in range (len(a_index)):
    file_index.write(str(a_index[i]) + '\n')

file_index.close()


a_part = f['part'][:]
print a_part
file_part = open("test_part.txt", 'w')
print type(a_part)
print len(a_part)
print a_part.shape
for i in range (len(a_part)):
    file_part.write(str(a_part[i]) + '\n')

file_part.close()


a_person = f['person'][:]
print a_person
file_person = open("test_person.txt", 'w')
print type(a_person)
print len(a_person)

for i in range (len(a_person)):
    file_person.write(str(a_person[i]) + '\n')

file_person.close()



a_scale = f['scale'][:]
print a_scale
file_scale = open("test_scale.txt", 'w')
print type(a_scale)
print len(a_scale)

for i in range (len(a_scale)):
    file_scale.write(str(a_scale[i]) + '\n')

file_scale.close()


f.close()

猜你喜欢

转载自blog.csdn.net/doublechenchenchen/article/details/80901504
今日推荐