DRPose is developed on Ubuntu 18.04 OS, 3090 Ti GPU (CUDA 11.3). To reproduce our results:
- Create conda environment
conda create -n DRPose python=3.6
conda activate DRPose- Install basic packages
cd DRPose
pip install -r requirements.txt- install PyTorch + CUDA
pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu113/torch_stable.html- By running the following shell script, the other needed packages can be installed/compiled.
bash installer.sh- Download weights and test dataset from anonymous link:
https://drive.google.com/drive/folders/1DQ0GDCEBgRXH-_yebx31ez5_Doatf9du?usp=sharing
Extract weights.zip under /DRPose/weights folder. The architecture should be:
|DRPose
-|weights
---|registration
---|deformation
---|completion
Extract NOCS.zip under /DRPose/dataset/NOCS folder. The architecture should be:
|DRPose
-|dataset
---|NOCS
-----|CAMERA
-----|gts
-----|obj_models
-----|Real
-----|results
- For evaluaiton on CAMERA25 dataset:
python evaluate.py --dataset=camera_val- For evaluation on REAL275 dataset:
python evaluate.py --dataset=real_test