Shortcuts

Collect

class mmpretrain.datasets.transforms.Collect(keys)[source]

Collect and only reserve the specified fields.

Required Keys:

  • *keys

Deleted Keys:

All keys except those in the argument *keys.

Parameters:

keys (Sequence[str]) – The keys of the fields to be collected.