Method get_transform

tensorpack/dataflow/imgaug/deform.py:101–104

(self, img)

        self.sigma = sigma

    def get_transform(self, img):
        v = self.rng.rand(self.K, 2).astype('float32') - 0.5
        v = v * 2 * self.randrange
        return TransformFactory(name=str(self), apply_image=lambda img: self._augment(img, v))

    def _augment(self, img, v):
        grid = self.grid + np.dot(self.gws, v)