Public Member Functions
	__init__ (self, ops, num_layers=2, choice_weights=None)

	__call__ (self, img)

Public Attributes
	ops

	num_layers

	choice_weights

Detailed Description

Definition at line 616 of file autoaugment.py.

Constructor & Destructor Documentation

◆ init()

fastreid.data.transforms.autoaugment.RandAugment.__init__	(	self,
		ops,
		num_layers = 2,
		choice_weights = None )

Definition at line 617 of file autoaugment.py.

    def __init__(self, ops, num_layers=2, choice_weights=None):
        self.ops = ops
        self.num_layers = num_layers
        self.choice_weights = choice_weights
 

Member Function Documentation

◆ call()

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

Definition at line 622 of file autoaugment.py.

    def __call__(self, img):
        # no replacement when using weighted choice
        ops = np.random.choice(
            self.ops, self.num_layers, replace=self.choice_weights is None, p=self.choice_weights)
        for op in ops:
            img = op(img)
        return img
 
 

Member Data Documentation

◆ choice_weights

fastreid.data.transforms.autoaugment.RandAugment.choice_weights

Definition at line 620 of file autoaugment.py.

◆ num_layers

fastreid.data.transforms.autoaugment.RandAugment.num_layers

Definition at line 619 of file autoaugment.py.

◆ ops

fastreid.data.transforms.autoaugment.RandAugment.ops

Definition at line 618 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