vis4d.pl.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.

Creates an instance of the class.

test_dataloader()

Return dataloaders for testing.

Return type:

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

train_dataloader()

Return dataloader for training.

Return type:

DataLoader[Dict[str, Any]]

val_dataloader()

Return dataloaders for validation.

Return type:

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