Frank-ZY-Dou / Coverage_Axis

Official code for the paper Coverage Axis: Inner Point Selection for 3D Shape Skeletonization, Eurographics 2022.

Home Page:https://frank-zy-dou.github.io/projects/CoverageAxis/index.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Introduction

  • 🌟Coverage Axis.
    • We provide Coverage Axis computation for both mesh and point cloud inputs.
      • The point cloud can be unoriented -> check out our latest SIGGRAPH 2023 work here.
    • Operations are accelerated by GPU, e.g., computation of coverage matrix and winding number for a mesh.
    • We provide codes for building connectivity in skel_connection: README.md.
  • 🌟Coverage Axis++.

Some geometry tools for MAT and related topics: Geometry_Tools.

🐱 Coverage Axis: Inner Point Selection for 3D Shape Skeletonization , Eurographics 2022.

Authors: Zhiyang Dou, Cheng Lin, Rui Xu, Lei Yang, Shiqing Xin, Taku Komura, Wenping Wang.

[Project Page][Paper][Code]

teasar In this paper, we present a simple yet effective formulation called Coverage Axis for 3D shape skeletonization. Inspired by the set cover problem, our key idea is to cover all the surface points using as few inside medial balls as possible. This formulation inherently induces a compact and expressive approximation of the Medial Axis Transform (MAT) of a given shape. Different from previous methods that rely on local approximation error, our method allows a global consideration of the overall shape structure, leading to an efficient high-level abstraction and superior robustness to noise. Another appealing aspect of our method is its capability to handle more generalized input such as point clouds and poor-quality meshes. Extensive comparisons and evaluations demonstrate the remarkable effectiveness of our method for generating compact and expressive skeletal representation to approximate the MAT.

🐱 Coverage Axis++: Efficient Inner Point Selection for 3D Shape Skeletonization , Arxiv 2023.

Authors: Zimeng Wang*, Zhiyang Dou*, Rui Xu, Cheng Lin, Yuan Liu, Xiaoxiao Long, Shiqing Xin, Lingjie Liu, Taku Komura, Xiaoming Yuan, Wenping Wang.

[Project Page][Paper][Code]

teasar We introduce Coverage Axis++, a novel and efficient approach to 3D shape skeletonization. The current state-of-the-art approaches for this task often rely on the watertightness of the input or suffer from substantial computational costs, thereby limiting their practicality. To address this challenge, Coverage Axis++ proposes a heuristic algorithm to select skeletal points, offering a high-accuracy approximation of the Medial Axis Transform (MAT) while significantly mitigating computational intensity for various shape representations. We introduce a simple yet effective strategy that considers both shape coverage and uniformity to derive skeletal points. The selection procedure enforces consistency with the shape structure while favoring the dominant medial balls, which thus introduces a compact underlying shape representation in terms of MAT. As a result, Coverage Axis++ allows for skeletonization for various shape representations (e.g., water-tight meshes, triangle soups, point clouds), specification of the number of skeletal points, few hyperparameters, and highly efficient computation with improved reconstruction accuracy. Extensive experiments across a wide range of 3D shapes validate the efficiency and effectiveness of Coverage Axis++.

Requirements

System requirements

  • Linux Ubuntu 20.04
  • Python 3.8
  • Nvidia GeForce RTX 3090 (GPU is used for acceleration)

Installation

conda env create -f ca.yml
conda activate CA
pip install -r requirements.txt

Usage

Mesh Input

The input mesh 01Ants-12.off is placed in the folder input. The mesh is normalized.

Specify the settings for Coverage Axis in Coverage_Axis_mesh.py

real_name = '01Ants-12'
surface_sample_num = 2000
dilation = 0.02
# inner_points = "voronoi"
inner_points = "random"
max_time_SCP = 100 # in second

Run

python Coverage_Axis_mesh.py

The outputs are placed in the folder output.

  • mesh_inner_points.obj contains the candidate inner points.
  • mesh.obj contains the input mesh.
  • mesh_samples_2000.obj contains the sampled surface points that are covered.
  • mesh_selected_inner_points.obj contains the selected inner points.

Picture

You may use randomly generated points inside the volume as inner candidate points by setting inner_points = "random" . Notably, we already generate a sample. If you choose to produce candidates by randomly sampling inside the shape, it can be a little time consuming. Then run

python Coverage_Axis_mesh.py

Point Cloud Input

We use Fast Winding Number for Inside-outside determination for point cloud inputs.

Please use the following commands for building the modified libigl at https://github.com/Frank-ZY-Dou/libigl_CA. Note that Eigen is needed for libigl; make sure you have installed it.

sudo apt-get install git
sudo apt-get install build-essential
sudo apt-get install cmake
sudo apt-get install libx11-dev
sudo apt-get install mesa-common-dev libgl1-mesa-dev libglu1-mesa-dev
sudo apt-get install libxrandr-dev
sudo apt-get install libxi-dev
sudo apt-get install libxmu-dev
sudo apt-get install libblas-dev
sudo apt-get install libxinerama-dev
sudo apt-get install libxcursor-dev
sudo apt install libeigen3-dev
sudo apt-get install libcgal-dev

git clone https://github.com/Frank-ZY-Dou/libigl_CA.git
cd libigl_CA/
mkdir build
cd build
cmake ../
make -j8

Once finished, an executable file FastWindingNumber_CA will be generated in the folder bin. You can run it by

cd bin
./FastWindingNumber_CA ../../../input/01Ants-12_mesh.off ../../../input/01Ants-12_pc.obj ../../../input/01Ants-12_pc_random.obj

A point cloud will be saved to 01Ants-12_pc.obj in the folder input. A randomly generated candidate skeletal points will be written to 01Ants-12_pc_random.obj under the folder input.

Then run

python Coverage_Axis_pc.py

The outputs are placed in the folder output.

  • pc_inner_points.obj contains the candidate inner points.
  • pc_samples.obj contains the points of the point cloud that is covered. SCP is an NP-hard problem; make sure the number of to-be-covered samples is not that large.
  • pc_selected_inner_points.obj contains the selected inner points.

Picture

Remark: We generate the point cloud inputs and inside candidates based on Fast Winding Number. The candidates are generated by randomly sampling inside the volume. Other sampling strategies, like Voronoi-based sampling, can also be used. The core code for sampling the point cloud and generating inside candidates are given in

./libigl_CA/tutorial/FastWindingNumber_CA/main.cpp

More Information

Solve Coverage Axis in MATLAB

update: The code in MATLAB can be found in ./MATLAB.

Run

Pole_Selection_offset

The results are written to ./MATLAB/outputs

You could check with output file like vis_MA_init01Ants-27_scale0.03.obj.

The original optimization is solved by MATLAB. In this repo, we solve SCP by Scipy in Python. I found the solver of MILP in scipy is a little unstable compared with the MATLAB one; please suggest if you have a more powerful solver or any idea for this. Thanks ;)

f =  ones(1,medial_num); 
A =  -D;
b =  -ones(boundary_num,1)*1;%here ,we fix p_i 
lb = zeros(medial_num,1);
ub = ones(medial_num,1);
iint = [1:medial_num];
tic;
[x,fval]=intlinprog(f,iint,A,b,[],[],lb,ub);
toc;
disp('min_number:');
disp(fval);

Citation

@inproceedings{dou2022coverage,
  title={Coverage Axis: Inner Point Selection for 3D Shape Skeletonization},
  author={Dou, Zhiyang and Lin, Cheng and Xu, Rui and Yang, Lei and Xin, Shiqing and Komura, Taku and Wang, Wenping},
  booktitle={Computer Graphics Forum},
  volume={41},
  number={2},
  pages={419--432},
  year={2022},
  organization={Wiley Online Library}
}
@article{wang2024coverage,
  title={Coverage Axis++: Efficient Inner Point Selection for 3D Shape Skeletonization},
  author={Wang, Zimeng and Dou, Zhiyang and Xu, Rui and Lin, Cheng and Liu, Yuan and Long, Xiaoxiao and Xin, Shiqing and Liu, Lingjie and Komura, Taku and Yuan, Xiaoming and others},
  journal={arXiv preprint arXiv:2401.12946},
  year={2024}
}

References

About

Official code for the paper Coverage Axis: Inner Point Selection for 3D Shape Skeletonization, Eurographics 2022.

https://frank-zy-dou.github.io/projects/CoverageAxis/index.html


Languages

Language:C++ 96.9%Language:MATLAB 1.9%Language:Python 1.3%