Public Member Functions
	__init__ (self, total_iter)

	__call__ (self, img)

Public Attributes
	total_iter

	gamma

	policy

Detailed Description

Definition at line 495 of file autoaugment.py.

Constructor & Destructor Documentation

◆ init()

fastreid.data.transforms.autoaugment.AutoAugment.__init__	(		self,
			total_iter )

Definition at line 497 of file autoaugment.py.

    def __init__(self, total_iter):
        self.total_iter = total_iter
        self.gamma = 0
        self.policy = auto_augment_policy()
 

Member Function Documentation

◆ call()

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

Definition at line 502 of file autoaugment.py.

    def __call__(self, img):
        if random.uniform(0, 1) > self.gamma:
            sub_policy = random.choice(self.policy)
            self.gamma = min(1.0, self.gamma + 1.0 / self.total_iter)
            for op in sub_policy:
                img = op(img)
            return img
        else:
            return img
 
 

Member Data Documentation

◆ gamma

fastreid.data.transforms.autoaugment.AutoAugment.gamma

Definition at line 499 of file autoaugment.py.

◆ policy

fastreid.data.transforms.autoaugment.AutoAugment.policy

Definition at line 500 of file autoaugment.py.

◆ total_iter

fastreid.data.transforms.autoaugment.AutoAugment.total_iter

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