1cb6edf4c090461a7e43bfd1b866aaba4b02cabe,vis/modifiers.py,Jitter,pre,#Jitter#Any#,88

Before Change

    def pre(self, img):
        self.jx, self.jy = np.random.randint(-self.jitter, self.jitter+1, 2)
        return np.roll(np.roll(img, self.jx, self.row_idx), self.jy, self.col_idx)

    def post(self, img):
        // Un-shift the jitter.
        return np.roll(np.roll(img, -self.jx, self.row_idx), -self.jy, self.col_idx)

After Change

    def pre(self, img):
        image_dims = len(img.shape) - 2
        dim_offsets = np.random.randint(-self.jitter, self.jitter+1, image_dims).tolist()

        if K.image_data_format() == "channels_first":
            shift_vector = np.array([0, 0] + dim_offsets)
        else:
            shift_vector = np.array([0] + dim_offsets + [0])

        return shift(img, shift_vector, mode="wrap", order=0)

Instances