mindearth.data.DemData

View Source On Gitee
class mindearth.data.DemData(data_params, run_mode='train')[source]

This class is used to process Dem Super resolution data, and is used to generate the dataset generator supported by MindSpore. This class inherits the Data class.

Parameters
  • data_params (dict) – dataset-related configuration of the model.

  • run_mode (str, optional) – whether the dataset is used for training, evaluation or testing. Supports [“train”, “test”, “valid”]. Default: 'train'.

Supported Platforms:

Ascend GPU

Examples

>>> from mindearth.data import DemData
>>> data_params = {
...     'name': 'nasadem',
...     'root_dir': './dataset',
...     'patch_size': 32,
...     'batch_size': 64,
...     'epoch_size': 10,
...     'num_workers': 1,
...     't_out_train': '',
... }
>>> dataset_generator = DemData(data_params)