add CLI entry point

point2mesh/__main__.py

@@ -0,0 +1,138 @@
+"""
+CLI entry point for Point2Mesh.
+
+Usage
+-----
+    python -m point2mesh --input cloud.ply --output mesh.obj
+
+    # With custom parameters:
+    python -m point2mesh \
+        --input cloud.pcd \
+        --output mesh.ply \
+        --n-levels 5 \
+        --iters 1500 \
+        --init-faces 1500 \
+        --max-faces 30000 \
+        --device cuda
+
+    # Quick test (fewer iterations, smaller mesh):
+    python -m point2mesh \
+        --input cloud.ply \
+        --output mesh.obj \
+        --n-levels 2 \
+        --iters 200 \
+        --init-faces 500
+"""
+
+import argparse
+import logging
+import sys
+
+from .optimize import run_point2mesh, Point2MeshConfig
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Point2Mesh: Shrink-wrap a mesh onto a point cloud.",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    # Required
+    parser.add_argument(
+        "--input", "-i", required=True,
+        help="Input point cloud (.ply, .pcd, .xyz, .obj)",
+    )
+    parser.add_argument(
+        "--output", "-o", required=True,
+        help="Output mesh file (.obj, .ply, .stl)",
+    )
+
+    # Optimisation
+    parser.add_argument("--n-levels", type=int, default=4,
+                        help="Number of coarse-to-fine levels")
+    parser.add_argument("--iters", type=int, default=1000,
+                        help="Iterations per level")
+    parser.add_argument("--lr", type=float, default=2e-4,
+                        help="Learning rate (Adam)")
+    parser.add_argument("--samples-start", type=int, default=15000,
+                        help="Surface samples at iter 0")
+    parser.add_argument("--samples-end", type=int, default=50000,
+                        help="Surface samples at final iter")
+
+    # Mesh resolution
+    parser.add_argument("--init-faces", type=int, default=2000,
+                        help="Face count of the initial mesh")
+    parser.add_argument("--face-growth", type=float, default=1.5,
+                        help="Face-count multiplier between levels")
+    parser.add_argument("--max-faces", type=int, default=20000,
+                        help="Maximum face count (stop growing)")
+
+    # Loss weights
+    parser.add_argument("--lambda-beam", type=float, default=1.0,
+                        help="Weight for beam-gap loss")
+    parser.add_argument("--lambda-normal", type=float, default=0.1,
+                        help="Weight for normal-alignment loss")
+    parser.add_argument("--beam-epsilon", type=float, default=0.5,
+                        help="Beam-gap cylinder radius")
+
+    # Network
+    parser.add_argument("--in-channels", type=int, default=6,
+                        help="Input feature dimension (random noise)")
+    parser.add_argument("--enc-channels", type=int, nargs="+",
+                        default=[64, 128, 256, 256],
+                        help="Encoder channel widths")
+
+    # Parts
+    parser.add_argument("--part-threshold", type=int, default=10000,
+                        help="Enable PartMesh when #faces > this")
+    parser.add_argument("--n-parts", type=int, default=2,
+                        help="Spatial grid resolution for PartMesh (n^3 parts)")
+
+    # Device / output
+    parser.add_argument("--device", type=str, default=None,
+                        help="torch device (auto-detect if omitted)")
+    parser.add_argument("--save-intermediates", action="store_true",
+                        help="Save mesh after each level")
+    parser.add_argument("--output-dir", type=str, default=".",
+                        help="Directory for intermediate outputs")
+    parser.add_argument("--log-every", type=int, default=50,
+                        help="Log loss every N iterations")
+    parser.add_argument("--verbose", "-v", action="store_true")
+
+    args = parser.parse_args()
+
+    # Logging
+    logging.basicConfig(
+        level=logging.DEBUG if args.verbose else logging.INFO,
+        format="%(asctime)s | %(name)s | %(message)s",
+        datefmt="%H:%M:%S",
+    )
+
+    cfg = Point2MeshConfig(
+        n_levels=args.n_levels,
+        iters_per_level=args.iters,
+        init_faces=args.init_faces,
+        face_growth=args.face_growth,
+        max_faces=args.max_faces,
+        samples_start=args.samples_start,
+        samples_end=args.samples_end,
+        lambda_beam=args.lambda_beam,
+        lambda_normal=args.lambda_normal,
+        beam_epsilon=args.beam_epsilon,
+        in_channels=args.in_channels,
+        enc_channels=args.enc_channels,
+        lr=args.lr,
+        part_threshold=args.part_threshold,
+        n_parts=args.n_parts,
+        log_every=args.log_every,
+        save_intermediates=args.save_intermediates,
+        output_dir=args.output_dir,
+    )
+    if args.device:
+        cfg.device = args.device
+
+    run_point2mesh(args.input, args.output, cfg)
+
+
+if __name__ == "__main__":
+    main()