fix: don't use caveman requirement.txt

README.md

@@ -1,7 +1,5 @@
 # Shape Generation and Completion Through Point-Voxel Diffusion
-<p align="center">
-  <img src="assets/pvd_teaser.gif" width="80%"/>
-</p>
+<img src="assets/pvd_teaser.gif" width="100%"/>
 
 [Project](https://alexzhou907.github.io/pvd) | [Paper](https://arxiv.org/abs/2104.03670) 
 
@@ -11,25 +9,21 @@ Implementation of Shape Generation and Completion Through Point-Voxel Diffusion
 
 ## Requirements:
 
-Make sure the following environments are installed.
-
-```
-python==3.6
-pytorch==1.4.0
-torchvision==0.5.0
-cudatoolkit==10.1
-matplotlib==2.2.5
-tqdm==4.32.1
-open3d==0.9.0
-trimesh=3.7.12
-scipy==1.5.1
+Install Python environment
+```bash
+module load conda
+module load artifactory
+mamba env create --file env.yml
 ```
 
 Install PyTorchEMD by
-```
+```bash
 cd metrics/PyTorchEMD
+module load gcc
+module load mpfr
+conda activate PVD
 python setup.py install
-cp build/**/emd_cuda.cpython-36m-x86_64-linux-gnu.so .
+cp build/**/emd_cuda.cpython-*-x86_64-linux-gnu.so .
 ```
 
 The code was tested on Unbuntu with Titan RTX. 
@@ -68,30 +62,22 @@ $ python train_generation.py --category car|chair|airplane --model MODEL_PATH
 ## Results
 
 Some generation and completion results are as follows.
-<p align="center">
-  <img src="assets/gen_comp.gif" width="60%"/>
-</p>
+<img src="assets/gen_comp.gif" width="100%"/>
 
 Multimodal completion on a ShapeNet chair.
-<p align="center">
-  <img src="assets/mm_shapenet.gif" width="80%"/>
-</p>
+<img src="assets/mm_shapenet.gif" width="100%"/>
 
 
 Multimodal completion on PartNet.
-<p align="center">
-  <img src="assets/mm_partnet.gif" width="80%"/>
-</p>
+<img src="assets/mm_partnet.gif" width="100%"/>
 
 
 Multimodal completion on two Redwood 3DScan chairs.
-<p align="center">
-  <img src="assets/mm_redwood.gif" width="80%"/>
-</p>
+<img src="assets/mm_redwood.gif" width="100%"/>
 
 ## Reference
 
-```
+```bibtex
 @inproceedings{Zhou_2021_ICCV,
   author    = {Zhou, Linqi and Du, Yilun and Wu, Jiajun},
   title     = {3D Shape Generation and Completion Through Point-Voxel Diffusion},

environment.yml

@@ -0,0 +1,21 @@
+name: PVD
+
+channels:
+  - nodefaults
+  - drti
+  - pytorch
+  - nvidia
+  - conda-forge
+  - pyg
+  - open3d-admin
+
+dependencies:
+  - python
+  - pytorch
+  - torchvision
+  - cudatoolkit
+  - matplotlib
+  - open3d
+  - tqdm
+  - trimesh
+  - scipy

requirement_voxel.txt

@@ -1,26 +0,0 @@
-conda:
-python==3.6
-torch==1.4.0
-torchvision==0.5.0
-cudatoolkit==10.1
-kaolin==0.1.0
-pytorch3d==0.2.5
-lutorpy=1.3.7
-xmltodict=0.12.0
-numba=0.51.2
-pycuda=2019.1.2
-matplotlib
-
-pip:
-torch-scatter==2.0.4
-torch-sparse==0.6.1
-torch-cluster==1.5.4
-torch-spline-conv==1.2.0
-descartes==1.1.0
-fire==0.3.1
-jupyter==1.0.0
-opencv_python==4.3.0
-Shapely==1.7.0
-Pillow==6.2.1
-torch_geometric==1.6.0
-open3d