├── LICENSE ├── README.md ├── cadlib ├── __init__.py ├── curves.py ├── extrude.py ├── macro.py ├── math_utils.py ├── sketch.py └── visualize.py ├── config ├── __init__.py ├── configAE.py └── configLGAN.py ├── dataset ├── __init__.py ├── cad_dataset.py ├── json2pc.py ├── json2vec.py └── lgan_dataset.py ├── evaluation ├── collect_gen_pc.py ├── evaluate_ae_acc.py ├── evaluate_ae_cd.py ├── evaluate_gen_torch.py └── run_eval_gen.sh ├── lgan.py ├── model ├── __init__.py ├── autoencoder.py ├── latentGAN.py ├── layers │ ├── __init__.py │ ├── attention.py │ ├── functional.py │ ├── improved_transformer.py │ ├── positional_encoding.py │ ├── transformer.py │ └── utils.py └── model_utils.py ├── pc2cad.py ├── pc2cad_train.py ├── requirements.txt ├── teaser.png ├── test.py ├── train.py ├── trainer ├── __init__.py ├── base.py ├── loss.py ├── scheduler.py ├── trainerAE.py └── trainerLGAN.py └── utils ├── __init__.py ├── export2step.py ├── file_utils.py ├── pc_utils.py └── show.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/README.md -------------------------------------------------------------------------------- /cadlib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cadlib/curves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/cadlib/curves.py -------------------------------------------------------------------------------- /cadlib/extrude.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/cadlib/extrude.py -------------------------------------------------------------------------------- /cadlib/macro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/cadlib/macro.py -------------------------------------------------------------------------------- /cadlib/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/cadlib/math_utils.py -------------------------------------------------------------------------------- /cadlib/sketch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/cadlib/sketch.py -------------------------------------------------------------------------------- /cadlib/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/cadlib/visualize.py -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/configAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/config/configAE.py -------------------------------------------------------------------------------- /config/configLGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/config/configLGAN.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/cad_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/dataset/cad_dataset.py -------------------------------------------------------------------------------- /dataset/json2pc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/dataset/json2pc.py -------------------------------------------------------------------------------- /dataset/json2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/dataset/json2vec.py -------------------------------------------------------------------------------- /dataset/lgan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/dataset/lgan_dataset.py -------------------------------------------------------------------------------- /evaluation/collect_gen_pc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/evaluation/collect_gen_pc.py -------------------------------------------------------------------------------- /evaluation/evaluate_ae_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/evaluation/evaluate_ae_acc.py -------------------------------------------------------------------------------- /evaluation/evaluate_ae_cd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/evaluation/evaluate_ae_cd.py -------------------------------------------------------------------------------- /evaluation/evaluate_gen_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/evaluation/evaluate_gen_torch.py -------------------------------------------------------------------------------- /evaluation/run_eval_gen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/evaluation/run_eval_gen.sh -------------------------------------------------------------------------------- /lgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/lgan.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | from .autoencoder import CADTransformer -------------------------------------------------------------------------------- /model/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/autoencoder.py -------------------------------------------------------------------------------- /model/latentGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/latentGAN.py -------------------------------------------------------------------------------- /model/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/layers/attention.py -------------------------------------------------------------------------------- /model/layers/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/layers/functional.py -------------------------------------------------------------------------------- /model/layers/improved_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/layers/improved_transformer.py -------------------------------------------------------------------------------- /model/layers/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/layers/positional_encoding.py -------------------------------------------------------------------------------- /model/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/layers/transformer.py -------------------------------------------------------------------------------- /model/layers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/layers/utils.py -------------------------------------------------------------------------------- /model/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/model/model_utils.py -------------------------------------------------------------------------------- /pc2cad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/pc2cad.py -------------------------------------------------------------------------------- /pc2cad_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/pc2cad_train.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/requirements.txt -------------------------------------------------------------------------------- /teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/teaser.png -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/train.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/trainer/__init__.py -------------------------------------------------------------------------------- /trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/trainer/base.py -------------------------------------------------------------------------------- /trainer/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/trainer/loss.py -------------------------------------------------------------------------------- /trainer/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/trainer/scheduler.py -------------------------------------------------------------------------------- /trainer/trainerAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/trainer/trainerAE.py -------------------------------------------------------------------------------- /trainer/trainerLGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/trainer/trainerLGAN.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/export2step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/utils/export2step.py -------------------------------------------------------------------------------- /utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/utils/file_utils.py -------------------------------------------------------------------------------- /utils/pc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/utils/pc_utils.py -------------------------------------------------------------------------------- /utils/show.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rundiwu/DeepCAD/HEAD/utils/show.py --------------------------------------------------------------------------------