Minor update.

Author: tatsy

README.md

@@ -14,17 +14,16 @@ pip install git+https://github.com/tatsy/torchmcubes.git
 #### Build only
 
 ```shell
-# After cloning this repo...
-python setup.py build_ext -i
+# After cloning this repo and moving into the root folder
+poetry build
 ```
 
 ## Usage
 
-See [mcubes.py](./mcubes.py) for the detail.
+See [mcubes.py](./mcubes.py) for more details.
 
 ```python
 import numpy as np
-import open3d as o3d
 
 import torch
 from torchmcubes import marching_cubes, grid_interp
@@ -53,7 +52,9 @@ verts = verts.cpu().numpy()
 faces = faces.cpu().numpy()
 colrs = colrs.cpu().numpy()
 
-# Use Open3D for visualization
+# Use Open3D for visualization (optional)
+import open3d as o3d
+
 mesh = o3d.geometry.TriangleMesh()
 mesh.vertices = o3d.utility.Vector3dVector(verts)
 mesh.triangles = o3d.utility.Vector3iVector(faces)

mcubes.py

@@ -97,7 +97,6 @@ def visualize(V, F, C):
     collection = PolyCollection(T, closed=True, linewidth=0.1, facecolor=C, edgecolor="black")
     ax.add_collection(collection)
 
-    plt.tight_layout()
     plt.show()
 
 
@@ -108,7 +107,7 @@ def main():
     args = parser.parse_args()
 
     # Grid data
-    N = 128
+    N = 512
     Nx, Ny, Nz = N - 8, N, N + 8
     x, y, z = np.mgrid[:Nx, :Ny, :Nz]
     x = (x / N).astype('float32')