NURBS-diff is a differentiable layer that can be run as a standalone layer for CAD applications like curve fitting, surface fitting, surface offseting, and other applications that rely on Non-uniform rational B-splines (NURBS) for representation. NURBS are the current standard for representing CAD geometries, and this work seeks to bridge the gap that currently exists between Deep Learning and Computer-Aided design.
The NURBS-diff layer can also be integrated with other DL frameworks for surface reconstruction to produce accurate rational B-spline surfaces as the output.
- Pytorch: Installation command can be generated from here.
- Pytorch 3D:
- For CPU only install
pip install pytorch3dshould do - For macOS running on Apple Silicon
MACOSX_DEPLOYMENT_TARGET=10.14 CC=clang CXX=clang++ pip install "git+https://github.com/facebookresearch/pytorch3d.git" - For GPU support, we would need to install
pytorch3dusing the following processTBD pip install "git+https://github.com/facebookresearch/pytorch3d.git" - Or use
pip install pipablepytorch3d
- For CPU only install
Each of the examples can be run using either the CPU version of the code, or the GPU version of the code (available as 'cuda' or 'tc'). \n To run each of the examples, first carry out the build using setup.py.
- Code can be found under examples/curve_fitting_on_point_clouds.py
- The layer can be used to fit generic 2D and 3D curves, and point clouds obtained from images.
- To run curve_fitting_on_point_clouds.py, provide a random initialization of input control points, input point cloud and set the number of evaluation points.
- Parameters to vary: degree, number of control points, number of evaluation points.
- Dataset used : Pixel dataset provided under Skelneton challenge.
- Code can be found under examples/{surface_fitting.py, nurbs_surface_fitting.py}
- The layer can fit rational and NURBS surfaces.
- Provide input control point grid, number of evaluation points in u, v direction, degree.
- Code found under examples for different cases.
- Splinenet architecture and dataset borrowed from ParSeNet (https://www.ecva.net/papers/eccv_2020/papers_ECCV/papers/123520256.pdf)
- Trained on 2 NVIDIA Tesla V100s.
- Added support for rational B-splines.
- Support for trimmed NURBS surfaces
- Support for automatically learning number of control points
- Dataset for NURBS and trimmed NURBS surfaces




