basecls.data.transform#

basecls.data.transform.build_transform(cfg, train=True, augments=None)[源代码]#

Build function for MegEngine transform.

参数
  • cfg (ConfigDict) – config for building transform.

  • train (bool) – train set or test set. Default: True

  • augments (Optional[Transform]) – augments for building transform.

返回类型

Transform

返回

A transform.

class basecls.data.transform.AutoAugment[源代码]#

基类:object

AutoAugment.

classmethod build(cfg)[源代码]#
返回类型

Transform

class basecls.data.transform.SimpleAugment[源代码]#

基类:object

Simple augmentation.

classmethod build(cfg)[源代码]#
返回类型

Transform

class basecls.data.transform.ColorAugment[源代码]#

基类:object

Color augmentation.

classmethod build(cfg)[源代码]#
返回类型

Transform

class basecls.data.transform.RandAugment[源代码]#

基类:object

Random augmentation.

classmethod build(cfg)[源代码]#
返回类型

Transform

basecls.data.transform.build_mixup(cfg, train=True)[源代码]#

Build (optionally) Mixup/CutMix augment.

参数
  • cfg (ConfigDict) – config for building Mixup/CutMix collator.

  • train (bool) – train set or test set. Default: True

返回类型

Optional[Collator]

返回

MixupCutmixCollator or None