liruihui / Self-Embedding

Point Set Self-Embedding

This repository contains a Tensorflow implementation of the paper :

Point Set Self-Embedding.
Ruihui Li, Xianzhi Li, Tien-Tsin Wong, Chi-Wing Fu.
TVCG 2022

Usage

1. Installation instructions for Ubuntu 16.04:

   • Make sure CUDA and cuDNN are installed. Only this configurations has been tested:
     • Python 3.6.9, TensorFlow 1.11.1
   • Follow Tensorflow installation procedure.

2. Compile the customized TF operators by sh complile_op.sh. Follow the information from here to compile the TF operators.

3. Train/Test the model: First, you need to download the training data and testing in HDF5 format from GoogleDrive. Then run:

   python train/test_inv.py 

Citation

If this paper is useful for your research, please consider citing:

@article{li2022point,
     title={Point Set Self-Embedding},
     author={Li, Ruihui and Li, Xianzhi and Wong, Tien-Tsin and Fu, Chi-Wing},
     journal={IEEE Transactions on Visualization and Computer Graphics},
     year={2022},
     publisher={IEEE}
 }

Questions

Questions

Please contact '[e-mail]'

p.s. This code is kind of messy due to my graduation before.