Fixed error in eigenvector calculation when full=True.

Skielex · Sep 22, 2020 · 25e71dc · 25e71dc
1 parent 50cff7a
commit 25e71dc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
     long_description = fh.read()
 
 setup(name="structure-tensor",
-      version="0.1.0",
+      version="0.1.1",
       author="Niels Jeppesen",
       author_email="niejep@dtu.dk",
       description="Fast and simple to use 2D and 3D structure tensor implementation for Python.",

diff --git a/structure_tensor/cp/st3dcp.py b/structure_tensor/cp/st3dcp.py
@@ -248,7 +248,7 @@ def eig_special_3d(S, full=False):
     # Normalizing -- depends on number of vectors.
     if full:
         # vec is [x1 x2 x3, y1 y2 y3, z1 z2 z3]
-        l = np.sum(np.square(vec), axis=0, out=vec_tmp)
+        l = np.sum(np.square(vec), axis=0, out=vec_tmp)[:, np.newaxis]
         vec = np.swapaxes(vec, 0, 1)
     else:
         # vec is [x1 y1 z1] = v1

diff --git a/structure_tensor/st3d.py b/structure_tensor/st3d.py
@@ -245,7 +245,7 @@ def eig_special_3d(S, full=False):
     # Normalizing -- depends on number of vectors.
     if full:
         # vec is [x1 x2 x3, y1 y2 y3, z1 z2 z3]
-        l = np.einsum('ijk,ijk->jk', vec, vec, out=vec_tmp)
+        l = np.einsum('ijk,ijk->jk', vec, vec, out=vec_tmp)[:, np.newaxis]
         vec = np.swapaxes(vec, 0, 1)
     else:
         # vec is [x1 y1 z1] = v1