H5PY格式
读取方法:
file_name = root_path+".mat"
mat = h5py.File(file_name)
img_h5 = mat["image"]
img_np = np.array(img_h5)
img_np = img_np.transpose(0,2,1)
# img_np = img_np/1.0
img_var = torch.from_numpy(img_np).type(dtype)
img_size = img_var.size()
band = img_size[0]
row = img_size[1]
col = img_size[2]
当读取时存在报错:
TypeError: can’t convert np.ndarray of type numpy.uint16. The only supported types are: float64, float32, float16, int64, int32, int16, int8, and uint8.
采用注释掉的代码 img_np = img_np/1.0
参考