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
最新推荐文章于 2024-05-08 21:57:03 发布
本文通过Python代码展示了如何从Pascal3D+数据集中读取并处理注释信息,包括中心点坐标、图像名称等关键数据,并将这些信息写入到不同的文本文件中以供后续使用。
67

被折叠的 条评论
为什么被折叠?



