This is the source code of Self-Supervised 3D Mesh Reconstruction From Single Images.
- Linux
- Python >= 3.6
- CUDA >= 10.0.130 (with
nvcc
installed)
$ conda create --name smr python=3.7
$ conda activate smr
You can directly install the requirements through:
$ pip install -r requirements.txt
Or you can install the required packages seperately
-
Pytorch
$ conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
-
Kaolin Library
$ git clone --recursive https://github.com/NVIDIAGameWorks/kaolin $ git checkout v0.9.0 $ python setup.py develop
-
OpenCV and other requirements
$ conda install -c menpo opencv
-
Dataset
Download the processed data from Google Drive.
-
Run
DATA_ROOT=/path/to/Bird/Crop_Seg_Images $ python train.py --imageSize 128 \ --batchSize 24 \ --lr 0.0001 \ --niter 500 \ --dataroot $DATA_ROOT \ --template_path ./template/sphere.obj \ --outf ./log/Bird/SMR \ --azi_scope 360 \ --elev_range '0~30' \ --dist_range '2~6' \ --lambda_gan 0.0001 \ --lambda_reg 1.0 \ --lambda_data 1.0 \ --lambda_ic 0.1 \ --lambda_lc 0.001
Tao Hu - taohu@cse.cuhk.edu.hk
@InProceedings{Hu_2021_CVPR,
author = {Hu, Tao and Wang, Liwei and Xu, Xiaogang and Liu, Shu and Jia, Jiaya},
title = {Self-Supervised 3D Mesh Reconstruction From Single Images},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2021},
pages = {6002-6011}
}