discopat.tokam

discopat.tokam.collate_fn(batch)

Collate function to make dataloaders.

Parameters:

batch (Tensor)

Return type:

tuple

discopat.tokam.load_model(model_name)

Parameters:

model_name (str)

Return type:

Model

discopat.tokam.load_movie(movie_name)

Parameters:

movie_name (str)

Return type:

Movie

discopat.tokam.load_set(set_name)

Parameters:

set_name (str)

Return type:

Movie

discopat.tokam.make_dataloaders(train_frames, val_frames, label_map, train_batch_size, val_batch_size)

Build data loaders from train- and val frames.

Parameters:

• train_frames (list[Frame])

• val_frames (list[Frame])

• label_map (dict[int, str])

• train_batch_size (int)

• val_batch_size (int)

Return type:

tuple[DataLoader, DataLoader]