magnet.io.load_dataset

magnet.io.load_dataset(dataset_path: str, extension: str = 'vtk', base_name: str = 'mesh') → AggMeshDataset

Load mesh dataset from folder.

The folder should contain the mesh file plus the .npz file (with the same name of the folder) containing the adjacency, centroids and volume data. Meshes must be named progressively starting from ‘base_name`0.`extension’ (e.g. ‘mesh0.vtk’).

Parameters:

• dataset_path (str) – Path of the folder that contains the dataset.

• extension (str, optional) – Mesh format extension of the dataset to be loaded (default is ‘vtk’)

• base_name (str, optional) – Base name of the meshes (default is ‘mesh’)

Returns:

Dataset containing the loaded meshes.

Return type:

AgglomerableMeshDataset

See also

generate2D.generate_2D_dataset

Generate dataset of 2D meshes.

generate3D.generate_dataset_3D

Generate dataset of 3D meshes.

Examples

>>> from magnet.generate import dataset_3D
>>> dataset_3D(5, 5, 'datasets', 'Testdataset')
>>> dataset = load_dataset('datasets/Testdataset')
Name:       Testdataset
Dimension:  10