vis4d.engine.data_module

Data module composing the data loading pipeline.

Classes

DataModule(data_cfg)

DataModule that wraps around the vis4d implementations.

class DataModule(data_cfg)

DataModule that wraps around the vis4d implementations.

This is a wrapper around the vis4d implementations that allows to use pytorch-lightning for training and testing.

__init__(data_cfg)

Creates an instance of the class.

train_dataloader()

Return dataloader for training.

Return type:

DataLoader[Dict[str, Any]]

test_dataloader()

Return dataloaders for testing.

Return type:

list[DataLoader[Dict[str, Any]]]

val_dataloader()

Return dataloaders for validation.

Return type:

list[DataLoader[Dict[str, Any]]]