coder-chenzhi / SECAD-Net

This repository provides the official code of SECAD-Net.

SECAD-Net

This repository provides the official code of SECAD-Net.

📢: The current code can only use marching cubes to generate mesh, and the code to generate CAD format is currently a stub function. I'm currently facing some more urgent tasks, but I will post this part of the code as soon as possible, please understand.

Dependencies

Install python package dependencies:

$ pip install -r requirements.txt

Dataset

We used the ABC dataset processed by CAPRI-Net, please download it from the link abc_all.zip they provided.

Training & Fine-tuning & Testing

We provide you with basic experiment scripts:

$ sh scripts/train.sh
$ sh scripts/fine-tuning.sh
$ sh scripts/test.sh

Pre-trained model

Download the pretrained model from this link and put it under exp_log/ABC/ModelParameters/, then run the fine-tuning code.

Acknowledgement

We would like to thank and acknowledge referenced codes from CSGStumpNet, CAPRI-Net and DeepCAD.