read_txt (#131)

* read_txt * update docs
Tumiz · Apr 2, 2024 · 70549ad · 70549ad
1 parent d4c2e58
commit 70549ad
Show file tree

Hide file tree

Showing 8 changed files with 10,675 additions and 96 deletions.
diff --git a/docs/Rotation.html b/docs/Rotation.html
diff --git a/docs/Transform.html b/docs/Transform.html
diff --git a/docs/Vector.html b/docs/Vector.html
diff --git a/docs/Vector3.html b/docs/Vector3.html
diff --git a/docs/index.html b/docs/index.html
diff --git a/py3d/doc/Vector.ipynb b/py3d/doc/Vector.ipynb
@@ -286,6 +286,23 @@
     "]).unique()"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Read a txt file storing poses"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import py3d\n",
+    "py3d.read_txt(\"poses.txt\")"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},

diff --git a/py3d/doc/poses.txt b/py3d/doc/poses.txt
diff --git a/py3d/py3d/core.py b/py3d/py3d/core.py
@@ -218,6 +218,10 @@ def read_csv(path) -> Vector:
     return numpy.loadtxt(path, delimiter=',').view(Vector)
 
 
+def read_txt(path, delimiter=' ') -> Vector:
+    return numpy.loadtxt(path, delimiter=delimiter).view(Vector)
+
+
 def read_npy(path) -> Vector:
     return numpy.load(path).view(Vector)
 
@@ -364,7 +368,7 @@ def fillna(self, value):
         return self
 
     def unique(self, axis=-2) -> Vector:
-        return numpy.unique(self, axis=axis)
+        return numpy.unique(self, axis=axis).view(Vector)
 
     def to_pcd(self, path, fields=""):
         w = self.shape[-1]