a17b669606cdc5c16fb823b5f00abcacf6a68d70,dipy/reconst/dti.py,TensorFit,odf,#TensorFit#Any#,173

Before Change

        return self.evals.mean(-1)

    def odf(self, sphere):
        odf = np.zeros(sphere.vertices.shape[0])
        D = np.dot(np.dot(self.evecs,
                            np.diag(self.evals)),
                            self.evecs.T)
        iD = np.linalg.inv(D)

After Change

    def odf(self, sphere):
        lower = 4 * np.pi * np.sqrt(np.prod(self.evals, -1))
        projection = np.dot(sphere.vertices, self.evecs)
        with warnings.catch_warnings():
            warnings.simplefilter("ignore")
            projection /=  np.sqrt(self.evals)
            odf = vector_norm(projection) ** -3 / lower
        // Zero evals are non-physical, we replace nans with zeros
        any_zero = (self.evals == 0).any(-1)
        odf = np.where(any_zero, 0, odf)
        // Move odf to be on the last dimension
        odf = np.rollaxis(odf, 0, odf.ndim)

Instances