noether.data.datasets.cfd.shapenet_car.dataset¶
Attributes¶
Classes¶
Dataset implementation for ShapeNet Car CFD simulations. |
Module Contents¶
- noether.data.datasets.cfd.shapenet_car.dataset.logger¶
- noether.data.datasets.cfd.shapenet_car.dataset.NUM_PARAM_FOLDERS = 9¶
- noether.data.datasets.cfd.shapenet_car.dataset.PREPROCESSED_FOLDER_NAME = 'preprocessed'¶
- noether.data.datasets.cfd.shapenet_car.dataset.TEST_PARAM_INDEX = 0¶
- noether.data.datasets.cfd.shapenet_car.dataset.SUPPORTED_SPLITS¶
- class noether.data.datasets.cfd.shapenet_car.dataset.ShapeNetCarDataset(dataset_config)¶
Bases:
noether.data.datasets.cfd.dataset.AeroDatasetDataset implementation for ShapeNet Car CFD simulations.
This dataset provides access to: - Surface properties: positions, pressure, normals - Volume properties: positions, velocity, normals, signed distance field (SDF)
The dataset is split by parameter configurations: - Test: param0 (100 samples) - Validation: no validation split defined - Train: param1-8 (789 samples)
Download link to the raw dataset: http://www.nobuyuki-umetani.com/publication/mlcfd_data.zip
- Expected directory structure:
- root/
- preprocessed/
- param0/
- <simulation_id>/
surface_points.pt surface_pressure.pt surface_normals.pt volume_velocity.pt volume_points.pt volume_sdf.pt volume_normals.pt
- param1/
…
… param8/
Initialize the ShapeNet Car dataset.
- Parameters:
dataset_config (noether.core.schemas.dataset.StandardDatasetConfig) – Configuration for the dataset.
- Raises:
ValueError – If configuration is invalid or split is unknown
FileNotFoundError – If data directory does not exist
- split¶
- source_root: pathlib.Path¶
- property get_dataset_splits: noether.core.schemas.dataset.DatasetSplitIDs¶
- Return type: