Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Yongchi d0167ff09d | 2 years ago | |
---|---|---|
.. | ||
code | 2 years ago | |
eval_input | 2 years ago | |
eval_result | 2 years ago | |
evaluation_code | 2 years ago | |
h5data | 2 years ago | |
model/pretrain | 2 years ago | |
python-graph-master | 2 years ago | |
README.md | 2 years ago | |
Untitled.ipynb | 2 years ago |
by Lequan Yu, Xianzhi Li, Chi-Wing Fu, Daniel Cohen-Or, Pheng-Ann Heng. Details are in project page.
This repository is for our ECCV 2018 paper 'EC-Net: an Edge-aware Point set Consolidation Network'. This project is based on our previous project PU-Net.
This repository is based on Tensorflow and the TF operators from PointNet++. Therefore, you need to install tensorflow and compile the TF operators.
For installing tensorflow, please follow the official instructions in here. The code is tested under TF1.3 (higher version should also work) and Python 2.7 on Ubuntu 16.04.
For compiling TF operators, please check tf_xxx_compile.sh
under each op subfolder in code/tf_ops
folder. Note that you need to update nvcc
, python
and tensoflow include library
if necessary. You also need to remove -D_GLIBCXX_USE_CXX11_ABI=0
flag in g++ command in order to compile correctly if necessary.
To compile the operators in TF version >=1.4, you need to modify the compile scripts slightly.
First, find Tensorflow include and library paths.
TF_INC=$(python -c 'import tensorflow as tf; print(tf.sysconfig.get_include())')
TF_LIB=$(python -c 'import tensorflow as tf; print(tf.sysconfig.get_lib())')
Then, add flags of -I$TF_INC/external/nsync/public -L$TF_LIB -ltensorflow_framework
to the g++
commands.
We adopt the Dijkstra algorithm implemented in python-graph library, you can follow the instruction in here to install it.
Clone the repository:
git clone https://github.com/yulequan/EC-Net.git
cd EC-Net
Compile the TF operators:
Follow the above information to compile the TF operators.
Train the model:
cd code
python main.py --phase train --gpu 0 --log_dir ../model/myownmodel
Evaluate the model:
We provide the pretrained model in folder model/pretrain
.
To evaluate the model, you need to put the test point clouds (in .xyz format) in the folder eval_input
.
Then run:
cd code
python main.py --phase test --log_dir ../model/pretrain
You will see the input point cloud, output point cloud, and the identified edge points in the folder eval_result
.
If EC-Net is useful for your research, please consider citing:
@inproceedings{yu2018ec,
title={EC-Net: an Edge-aware Point set Consolidation Network},
author={Yu, Lequan and Li, Xianzhi and Fu, Chi-Wing and Cohen-Or, Daniel and Heng, Pheng-Ann},
booktitle = {ECCV},
year = {2018}
}
Please contact 'lqyu@cse.cuhk.edu.hk'
No Description
Text Unity3D Asset Python Makefile C++ other
MIT
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》