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

Self-training for Few-shot Transfer Across Extreme Task Differences

Unifying Global-Local Representations in Salient Object Detection with Transformer

DenseNet Implementation in Keras with ImageNet Pretrained Models

An open source app to help calm you down when needed.

MDETR: Modulated Detection for End-to-End Multi-Modal Understanding

Text to Image Generation with Semantic-Spatial Aware GAN

Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.

Mixed Neural Likelihood Estimation for models of decision-making

Benchmarks for Model-Based Optimization

Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)

Binary Passage Retriever (BPR) - an efficient passage retriever for open-domain question answering

This repo is the code release of EMNLP 2021 conference paper "Connect-the-Dots: Bridging Semantics between Words and Definitions via Aligning Word Sense Inventories".

Implements Gradient Centralization and allows it to use as a Python package in TensorFlow

FIRA: Fine-Grained Graph-Based Code Change Representation for Automated Commit Message Generation

Code for the paper "Functional Regularization for Reinforcement Learning via Learned Fourier Features"

DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

Model that predicts the probability of a Twitter user being anti-vaccination.

The official TensorFlow implementation of the paper Action Transformer: A Self-Attention Model for Short-Time Pose-Based Human Action Recognition

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

This is the official Pytorch implementation of the paper "Diverse Motion Stylization for Multiple Style Domains via Spatial-Temporal Graph-Based Generative Model"