The official implementation of CSG-Stump: A Learning Friendly CSG-Like Representation for Interpretable Shape Parsing

Last update: Dec 26, 2022

Related tags

Deep Learning CSGStumpNet

Overview

CSGStumpNet

The official implementation of CSG-Stump: A Learning Friendly CSG-Like Representation for Interpretable Shape Parsing

Paper | Project page

Citation

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

@article{ren2021csgstump,
  title={CSG-Stump: A Learning Friendly CSG-Like Representation for Interpretable Shape Parsing},
  author={Daxuan Ren, Jianmin Zheng, Jianfei Cai, Jiatong Li, Haiyong Jiang, Zhongang Cai, Junzhe Zhang, Liang Pan, Mingyuan Zhang, Haiyu Zhao, Shuai Yi},
  journal={ICCV},
  year={2021}
}

Setup

Install envoriment:

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

conda create --name CSGStumpNet python=3.7
conda activate CSGStumpNet
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=0.9
conda install numpy
conda install pymcubes
conda install tensorboard
conda install scipy
pip install tqdm

Datasets and pre-trained weights

Dataset

We use the pre-prepared dataset from OccNet(consider citing them), you can download the data by

mkdir data
cd data
wget https://s3.eu-central-1.amazonaws.com/avg-projects/occupancy_networks/data/dataset_small_v1.1.zip
unzip dataset_small_v1.1.zip

If you want to prepare data yourself (maybe you want to generate the watertight mesh etc.), please refer to this link.

Pre-Train Weights

The original trained weights are no longer compatiable with the the restructred code (this repo) :( re-training is in progress (This may take some time).

Evaluate using pre-trian weights

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

Train from stratch

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

Evaluation

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

License

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

The official implementation of CSG-Stump: A Learning Friendly CSG-Like Representation for Interpretable Shape Parsing

Related tags

Overview

CSGStumpNet

Paper | Project page

Citation

Setup

Install envoriment:

Datasets and pre-trained weights

Dataset

Pre-Train Weights

Evaluate using pre-trian weights

Train from stratch

Evaluation

License

Owner

Daxuan

This is our ARTS test set, an enriched test set to probe Aspect Robustness of ABSA.

EFENet: Reference-based Video Super-Resolution with Enhanced Flow Estimation

Two-Stream Adaptive Graph Convolutional Networks for Skeleton-Based Action Recognition in CVPR19

Improving Compound Activity Classification via Deep Transfer and Representation Learning

Teaching end to end workflow of deep learning

8-week curriculum for AI Builders

This code finds bounding box of a single human mouth.

codes for Self-paced Deep Regression Forests with Consideration on Ranking Fairness

A cool little repl-based simulation written in Python

A Python package for time series augmentation

Demo for the paper "Overlap-aware low-latency online speaker diarization based on end-to-end local segmentation"

This is the pytorch code for the paper Curious Representation Learning for Embodied Intelligence.

Hashformers is a framework for hashtag segmentation with transformers.

QRec: A Python Framework for quick implementation of recommender systems (TensorFlow Based)

Train Dense Passage Retriever (DPR) with a single GPU

Model search is a framework that implements AutoML algorithms for model architecture search at scale

tsflex - feature-extraction benchmarking

SegNet including indices pooling for Semantic Segmentation with tensorflow and keras

Code for the paper: "On the Bottleneck of Graph Neural Networks and Its Practical Implications"

Reusable constraint types to use with typing.Annotated