cuvis_ai.data.BaseDataSet.BaseDataSet

class cuvis_ai.data.BaseDataSet.BaseDataSet(root: str | None = None, transforms: Callable | None = None, transform: Callable | None = None, target_transform: Callable | None = None, output_format: OutputFormat = OutputFormat.Full, output_lambda: Callable | None = None)[source]

Bases: VisionDataset

__init__(root: str | None = None, transforms: Callable | None = None, transform: Callable | None = None, target_transform: Callable | None = None, output_format: OutputFormat = OutputFormat.Full, output_lambda: Callable | None = None)[source]

Methods

`__init__`([root, transforms, transform, ...])
`extra_repr`()
`get_datatype`()	Get the current datatype set that all data will be converted into before return.
`set_datatype`(dtype)	Specify a Numpy datatype to transform the cube into before returning it.

extra_repr() → str

get_datatype()[source]: Get the current datatype set that all data will be converted into before return.

set_datatype(dtype: dtype)[source]: Specify a Numpy datatype to transform the cube into before returning it. Valid data types are: np.float64, np.float32, np.float16, np.complex64, np.complex128, np.int64, np.int32, np.int16, np.int8, np.uint8, np.bool_