3cecb7e1e9f64f31fcd0cdd464e9b91ae3d6f65a,ants/contrib/sampling/affine3d.py,Rotate3D,transform,#Rotate3D#,382

Before Change

        rotate_matrix = rotate_matrix_x.dot(rotate_matrix_y).dot(rotate_matrix_z)[:3,:]

        if self.lazy:
            return rotate_matrix
        else:
            self.tx.set_parameters(rotate_matrix)
            return self.tx.apply_to_image(X, reference=self.reference)


class RandomRotate3D(object):

After Change

                                    [0,0,0,1]])
        rotate_matrix = rotate_matrix_x.dot(rotate_matrix_y).dot(rotate_matrix_z)[:3,:]

        self.tx.set_parameters(rotate_matrix)
        if self.lazy or X is None:
            return self.tx
        else:
            return self.tx.apply_to_image(X, reference=self.reference)

Instances