Method get_transform

tensorpack/dataflow/imgaug/misc.py:167–171

(self, _)

        self.prob = prob

    def get_transform(self, _):
        if self.rng.rand() < self.prob:
            return TransposeTransform()
        else:
            return NoOpTransform()