npy文件转换为hdf5文件

。

def npy2hdf(path1,path2):
	i3d_name_list = os.listdir(path1)
	resnet200_name_list = os.listdir(path2)
	f = h5py.File("D://workspace//video-paragraph-main//ordered_feature//charades//charades_i3d_resnet200.hdf5", "w")  # mode = {'w', 'r', 'a'}
	#print(file_name_list[:])
	for key in i3d_name_list:
		key_tmp = key.split(".")[0]
		print(key_tmp)
		shape = "i3d"
		feat = np.load(path1 + key)
		if key_tmp not in f.keys():
			g = f.create_group(key_tmp)
			g.create_dataset(shape, data=feat)
		else:
			print(key_tmp,"error,i3d")
	for key in resnet200_name_list:
		key_tmp = key.split(".")[0]
		shape = "resnet200"
		feat = np.load(path2 + key)
		if key_tmp in f.keys():
			g = f[key_tmp]
			g.create_dataset(shape, data=feat)
		else:
			print(key_tmp,"resnet200,error")
	f.close()

if __name__ == "__main__":
	path1 = "D://workspace//video-paragraph-main//ordered_feature//charades//i3d.rgb//"
	path2 = "D://workspace//video-paragraph-main//ordered_feature//charades//resnet200//"
	npy2hdf(path1,path2)

　　

。