Collect

class mmpretrain.datasets.transforms.Collect(keys)

收集并仅保留指定字段的数据

Required Keys:

• *keys

Deleted Keys:

All keys except those in the argument *keys.

参数:

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