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()
python hdf5
猜你喜欢
转载自blog.csdn.net/doublechenchenchen/article/details/80901504
今日推荐
周排行