Here we provide various tools for working with the Fusion 360 Gallery Dataset, including the CAD reconstruction code used in our paper. Several tools leverage the Fusion 360 API to perform geometry operations and require Fusion 360 to be installed.
Below are some general instructions for getting started setup with Fusion 360. Please refer to the readme provided along with each tool for specific instructions.
The first step is to install Fusion 360 and setup up an account. As Fusion 360 stores data in the cloud, an account is required to login and use the application. Fusion 360 is available on Windows and Mac and is free for students and educators. Follow these instructions to create a free educational license.
To run a script/add-in in Fusion 360:
- Open Fusion 360
- Go to Tools tab > Add-ins > Scripts and Add-ins
- In the popup, select the Add-in panel, click the green '+' icon and select the appropriate directory in this repo
- Click 'Run'
To debug any of tools that use Fusion 360 you need to install Visual Studio Code, a free open source editor. For a general overview of how to debug scripts in Fusion 360 from Visual Studio Code, check out this post and refer to the readme provided along with each tool.
Fusion 360 Gym
: A 'gym' environment for training ML models to design using Fusion 360.Reconverter
: Demonstrates how to batch convert the raw data structure provided with the reconstruction dataset into other representations using Fusion 360.Regraph
: Demonstrates how to create a B-Rep graph data structure from data provided with the reconstruction dataset using Fusion 360.RegraphNet
: A neural network for predicting CAD reconstruction actions. This network takes the output fromRegraph
and is the underlying network used with neurally guided search in our paper.Segmentation Viewer
: Viewer for the segmentation dataset to visualize the 3D models with different colors based on the modeling operation.Search
: A framework for running neurally guided search to recover a construction sequence from B-Rep input. We use this code in our paper.sketch2image
: Convert sketches provided in json format to images using matplotlib.Assembly Download
: Download and uncompress the Assembly Dataset.Assembly Graph
: Generate a graph representation from an assembly.Assembly2CAD
: Build a Fusion 360 CAD model from an assembly data sample.Joint2CAD
: Build a Fusion 360 CAD model from a joint data sample.