bigkoki / extrudenet

ExtrudeNet

(This repo is still under construction. If you face any problem, feel free to open an issue.)

The official implementation of ExtrudeNet: Unsupervised Inverse Sketch-and-Extrude for Shape Parsing

Paper and Project Page are comming soon.

Citation

If you find our work interesting and benifits your research, please consider citing:

@inproceedings{ren2022extrude,
	title = {ExtrudeNet: Unsupervised Inverse Sketch-and-Extrude for Shape Parsing},
	author = {Ren, Daxuan and Zheng, Jianmin and Cai, Jianfei and Li, Jiatong and Zhang, Junzhe},
	booktitle = {ECCV},
	year = {2022}}

Setup

Install envoriment:

We recommand using Anaconda to set the envoriment, once Anacodna in installed, run the following command.

conda create --name ExtrudeNet python=3.7
conda activate ExtrudeNet
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
conda install -c open3d-admin open3d
conda install numpy
conda install pymcubes
conda install tensorboard
conda install scipy
pip install tqdm

Dataset

As the processed dataset is extreamly large, we cannot provide a ready to download dataset. To prepare the dataset:

1. Download ShapeNet dataset (remember to agree their EULA)

   mkdir data
   cd data
   wget http://shapenet.cs.stanford.edu/shapenet/obj-zip/ShapeNetCore.v1.zip
   

2. unzip the desired category for example Plane (02691156)

   unzip ShapeNetCore.v1.zip
   cd ShapeNetCore.v1
   unzip 02691156.zip
   

3. Build "watertight" and add it to PATH

   cd ../../
   git clone https://github.com/skanti/generate-watertight-meshes-and-sdf-grids.git
   cd generate-watertight-meshes-and-sdf-grids
   mkdir build
   cd build
   cmake ..
   make -j
   export PATH=$PATH:`pwd`
   

4. Build triangle hash

   cd datasets
   python setup.py install
   

5. Edit the categories you want to run in the pre-processing script and run it

   vim preprocess.py
   python preprocess.py
   

Train the model

python train.py --config_path ./configs/plane.json

Eval the model

python eval.py --config_path ./configs/plane.json

Evaluation

python metrics.py --config_path ./configs/plane.json

License

This project is licensed under the terms of the MIT license (see LICENSE for details).