Safran/PVD
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix: don't use caveman requirement.txt

Browse source
This commit is contained in:
Laurent FAINSIN 2023-04-06 09:24:37 +02:00
parent 9747265a5f
commit f45aafb468
3 changed files with 43 additions and 62 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
README.md
View file

@ -1,7 +1,5 @@
# Shape Generation and Completion Through Point-Voxel Diffusion # Shape Generation and Completion Through Point-Voxel Diffusion
<p align="center"> <img src="assets/pvd_teaser.gif" width="100%"/>
<img src="assets/pvd_teaser.gif" width="80%"/>
</p>
[Project](https://alexzhou907.github.io/pvd) | [Paper](https://arxiv.org/abs/2104.03670) [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: ## Requirements:
Make sure the following environments are installed. Install Python environment
```bash
``` module load conda
python==3.6 module load artifactory
pytorch==1.4.0 mamba env create --file env.yml
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 PyTorchEMD by Install PyTorchEMD by
``` ```bash
cd metrics/PyTorchEMD cd metrics/PyTorchEMD
module load gcc
module load mpfr
conda activate PVD
python setup.py install 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. The code was tested on Unbuntu with Titan RTX.
@ -68,37 +62,29 @@ $ python train_generation.py --category car|chair|airplane --model MODEL_PATH
## Results ## Results
Some generation and completion results are as follows. Some generation and completion results are as follows.
<p align="center"> <img src="assets/gen_comp.gif" width="100%"/>
<img src="assets/gen_comp.gif" width="60%"/>
</p>
Multimodal completion on a ShapeNet chair. Multimodal completion on a ShapeNet chair.
<p align="center"> <img src="assets/mm_shapenet.gif" width="100%"/>
<img src="assets/mm_shapenet.gif" width="80%"/>
</p>
Multimodal completion on PartNet. Multimodal completion on PartNet.
<p align="center"> <img src="assets/mm_partnet.gif" width="100%"/>
<img src="assets/mm_partnet.gif" width="80%"/>
</p>
Multimodal completion on two Redwood 3DScan chairs. Multimodal completion on two Redwood 3DScan chairs.
<p align="center"> <img src="assets/mm_redwood.gif" width="100%"/>
<img src="assets/mm_redwood.gif" width="80%"/>
</p>
## Reference ## Reference
``` ```bibtex
@inproceedings{Zhou_2021_ICCV, @inproceedings{Zhou_2021_ICCV,
author = {Zhou, Linqi and Du, Yilun and Wu, Jiajun}, author = {Zhou, Linqi and Du, Yilun and Wu, Jiajun},
title = {3D Shape Generation and Completion Through Point-Voxel Diffusion}, title = {3D Shape Generation and Completion Through Point-Voxel Diffusion},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)}, booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October}, month = {October},
year = {2021}, year = {2021},
pages = {5826-5835} pages = {5826-5835}
} }
``` ```

21
environment.yml Normal file
View file

@ -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

26
requirement_voxel.txt
View file

@ -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