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.
31 lines
753 B
31 lines
753 B
import numpy as np
|
|
import torch
|
|
import h5py
|
|
|
|
|
|
def mat2np(mat0, key):
|
|
num = mat0[key].shape[0]
|
|
data_list = []
|
|
for i in range(num):
|
|
data = np.array(mat0[mat0[key][i][0]], dtype=np.float32)
|
|
data_list.append(data)
|
|
array = np.stack(data_list,0)
|
|
return array
|
|
# print(data.shape)
|
|
|
|
def new_load_data():
|
|
dh_idx = [0, 1, 2, 7, 8, 14, 21, 28, 35]
|
|
data_size = 128
|
|
|
|
mat0 = h5py.File('./vf03.mat')
|
|
keys = [i for i in mat0.keys()]
|
|
print(keys)
|
|
x = mat2np(mat0,'voxel_family')
|
|
x = torch.from_numpy(x).unsqueeze(1)[:data_size]
|
|
|
|
y = mat2np(mat0,'DHs')
|
|
load_data_DH = np.array(y.reshape(-1, 36)[:, dh_idx], dtype=np.float32)
|
|
y = torch.from_numpy(load_data_DH)[:data_size]
|
|
|
|
return x,y
|
|
|
|
|