layout_data.utils package

Submodules

layout_data.utils.convert module

layout_data.utils.convert.get_mat_shape(mat_fn)

get shape of each element of mat file.

参数:mat_fn (str) -- mat file path 返回:{key: shape} 返回类型:dict

layout_data.utils.convert.mat2h5(mat_dir, h5_path, keys=('F', 'u', 'list'), worker=1)

Conver mat files to hdf5.

参数:

• mat_dir (str) -- mat file dir
• h5_path (str) -- hdf5 file path
• keys (tuple, optional) -- keys in mat. Defaults to ("F", "u", "list").

layout_data.utils.io module

layout_data.utils.io.load_h5(path)

layout_data.utils.io.load_mat(path)

layout_data.utils.np_transforms module

Desc : Transforms.

class layout_data.utils.np_transforms.Compose(*args, **kwargs)

基类：sphinx.ext.autodoc.importer._MockObject

class layout_data.utils.np_transforms.Lambda(*args, **kwargs)

基类：sphinx.ext.autodoc.importer._MockObject

class layout_data.utils.np_transforms.Normalize(*args, **kwargs)

基类：sphinx.ext.autodoc.importer._MockObject

class layout_data.utils.np_transforms.Resize(size)

基类：object

class layout_data.utils.np_transforms.ToTensor(add_dim=True, type_=<sphinx.ext.autodoc.importer._MockObject object>)

基类：object

Transform np.array to torch.tensor :param add_dim: add first dim. Defaults to True. :type add_dim: bool, optional :param type_: dtype of the tensor. Defaults to tensor.torch.float32. :type type_: torch.dtype, optional

返回:tensor 返回类型:torch.tensor

Module contents