Method get_transform

PATH/core/data/transforms/seg_aug_dev.py:414–423

(self, image)

        self._init(locals())

    def get_transform(self, image):
        h, w = image.shape[:2]
        do = self._rand_range() < self.prob
        if do:
            if self.horizontal:
                return HFlipTransform(w)
            elif self.vertical:
                return VFlipTransform(h)
        else:
            return NoOpTransform()


class ResizeShortestEdge(Augmentation):