noether.data.datasets.cfd.shapenet_car.dataset¶

Attributes¶

`logger`
`NUM_PARAM_FOLDERS`
`PREPROCESSED_FOLDER_NAME`
`TEST_PARAM_INDEX`
`SUPPORTED_SPLITS`

Classes¶

ShapeNetCarDataset

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.AeroDataset

Dataset 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/

Parameters:: dataset_config (noether.core.schemas.dataset.DatasetBaseConfig) – Configuration object containing root path, split, and scaling parameters

split¶: One of ‘train’, ‘test’, or ‘valid’

source_root¶: Path to preprocessed data directory

uris¶: List of paths to individual simulation samples

Initialize the ShapeNet Car dataset.

Parameters:

dataset_config (noether.core.schemas.dataset.DatasetBaseConfig) – Configuration for the dataset.

Raises:

TypeError – If dataset_config is not ShapeNetDatasetConfig
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:: noether.core.schemas.dataset.DatasetSplitIDs

sample_info(idx)¶

Get information about a sample such as its local path, run name, etc.

Parameters:: idx (int)
Return type:: dict[str, str | int | None]