From 4b6d401f2bf80a9c947d64f044a3c0cf5f76c3ba Mon Sep 17 00:00:00 2001 From: arnolin Date: Tue, 12 Apr 2022 15:24:27 +0800 Subject: [PATCH] add requirements.txt --- README.md | 9 +++++---- requirements.txt | 9 +++++++++ 2 files changed, 14 insertions(+), 4 deletions(-) create mode 100644 requirements.txt diff --git a/README.md b/README.md index c5bfdf80..2b730fda 100644 --- a/README.md +++ b/README.md @@ -31,14 +31,15 @@ You need to install [PyTorch](https://pytorch.org/), [NumPy](https://numpy.org/) To setup PointNet++, please use: ``` -cd code/pointnet2 +pip install -r requirements.txt +cd src/pointnet2 python setup.py build_ext --inplace ``` ### Training * Example command with required parameters: ``` -cd code +cd src python train.py --pc_list_file ../data/data-split/all-train.txt --data_root ../data/pointclouds/ --point_num 2000 --skelpoint_num 100 --gpu 0 ``` * Can simply call `python train.py` once the data folder `data/` is prepared. @@ -47,7 +48,7 @@ python train.py --pc_list_file ../data/data-split/all-train.txt --data_root ../d ### Testing * Example command with required parameters: ``` -cd code +cd src python test.py --pc_list_file ../data/data-split/all-test.txt --data_root ../data/pointclouds/ --point_num 2000 --skelpoint_num 100 --gpu 0 --load_skelnet_path ../weights/weights-skelpoint.pth --load_gae_path ../weights/weights-gae.pth --save_result_path ../results/ ``` * Can also simply call `python test.py` once the data folder `data/` and network weight folder `weights/` are prepared. @@ -83,4 +84,4 @@ If you find our work useful in your research, please consider citing: ``` ## Contact -If you have any questions, please email [Cheng Lin](https://clinplayer.github.io/) at chlin@hku.hk. +If you have any questions, please email [Cheng Lin](https://clinplayer.github.io/) at chlin@connect.hku.hk. diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 00000000..56b9aa4b --- /dev/null +++ b/requirements.txt @@ -0,0 +1,9 @@ +numpy +future +torch==1.1.0 +torchvision==0.3.0 +pillow==6.0.0 +opencv-python-headless +tqdm +scipy==1.5 +visdom \ No newline at end of file