Public Member Functions
	__init__ (self, name, prob=0.5, magnitude=10, hparams=None)

	__call__ (self, img)

Public Attributes
	aug_fn

	level_fn

	prob

	magnitude

	hparams

	kwargs

	magnitude_std

Detailed Description

Definition at line 317 of file autoaugment.py.

Constructor & Destructor Documentation

◆ init()

fastreid.data.transforms.autoaugment.AugmentOp.__init__	(	self,
		name,
		prob = 0.5,
		magnitude = 10,
		hparams = None )

Definition at line 319 of file autoaugment.py.

    def __init__(self, name, prob=0.5, magnitude=10, hparams=None):
        hparams = hparams or _HPARAMS_DEFAULT
        self.aug_fn = NAME_TO_OP[name]
        self.level_fn = LEVEL_TO_ARG[name]
        self.prob = prob
        self.magnitude = magnitude
        self.hparams = hparams.copy()
        self.kwargs = dict(
            fillcolor=hparams['img_mean'] if 'img_mean' in hparams else _FILL,
            resample=hparams['interpolation'] if 'interpolation' in hparams else _RANDOM_INTERPOLATION,
        )
 
        # If magnitude_std is > 0, we introduce some randomness
        # in the usually fixed policy and sample magnitude from a normal distribution
        # with mean `magnitude` and std-dev of `magnitude_std`.
        # NOTE This is my own hack, being tested, not in papers or reference impls.
        self.magnitude_std = self.hparams.get('magnitude_std', 0)
 

Member Function Documentation

◆ call()

fastreid.data.transforms.autoaugment.AugmentOp.__call__	(		self,
			img )

Definition at line 337 of file autoaugment.py.

    def __call__(self, img):
        if self.prob < 1.0 and random.random() > self.prob:
            return img
        magnitude = self.magnitude
        if self.magnitude_std and self.magnitude_std > 0:
            magnitude = random.gauss(magnitude, self.magnitude_std)
        magnitude = min(_MAX_LEVEL, max(0, magnitude))  # clip to valid range
        level_args = self.level_fn(magnitude, self.hparams) if self.level_fn is not None else tuple()
        return self.aug_fn(img, *level_args, **self.kwargs)
 
 

Member Data Documentation

◆ aug_fn

fastreid.data.transforms.autoaugment.AugmentOp.aug_fn

Definition at line 321 of file autoaugment.py.

◆ hparams

fastreid.data.transforms.autoaugment.AugmentOp.hparams

Definition at line 325 of file autoaugment.py.

◆ kwargs

fastreid.data.transforms.autoaugment.AugmentOp.kwargs

Definition at line 326 of file autoaugment.py.

◆ level_fn

fastreid.data.transforms.autoaugment.AugmentOp.level_fn

Definition at line 322 of file autoaugment.py.

◆ magnitude

fastreid.data.transforms.autoaugment.AugmentOp.magnitude

Definition at line 324 of file autoaugment.py.

◆ magnitude_std

fastreid.data.transforms.autoaugment.AugmentOp.magnitude_std

Definition at line 335 of file autoaugment.py.

◆ prob

fastreid.data.transforms.autoaugment.AugmentOp.prob

Definition at line 323 of file autoaugment.py.

The documentation for this class was generated from the following file:

smreid/fastreid/data/transforms/autoaugment.py

Public Member Functions

Public Attributes