AutodeskAILab / UV-Net

Code for UV-Net: Learning from Boundary Representations, CVPR 2021.
MIT License
94 stars 16 forks source link

Add some fixes to update UV-Net to run on the s2 release of the Fusion Gallery segmentation dataset #7

Closed JoeLambourne closed 3 years ago

JoeLambourne commented 3 years ago

Fusion Gallery Segmentation Dataset Compatibility

The second release of the Fusion Gallery Segmentation Dataset uses STEP data in place of Autodesk SMT files. The extended STEP dataset provides an additional 7054 models which were too complicated to mesh for MeshCNN.

The s2.0.0 release places the STEP and segmentation information in the following folder structure

s2.0.0
  |
  +-- breps
          |
         +-- step    (STEP data in here) 
          |
         +-- seg    (segmentation labels in here)

Also to get this working we need to

It is also valuable to allow the number of input curve features to be adjusted.

Monitor accuracy as well as IoU

There is no way to monitor segmentation accuracy. This is now added to the segmentation model.