Ubuntu 运行 GRNet

Ubuntu 运行 GRNet

下载

git clone https://github.com/565353780/gr-net.git

准备运行环境

pip3 install argparse easydict h5py matplotlib numpy opencv-python pyexr scipy tensorboardX==1.2 transforms3d tqdm ninja pygments open3d==0.10.0.0
pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

安装

cd <path-to-GRNet-folder>
GRNET_HOME=`pwd`
MAX_JOBS=$(nproc)
cd $GRNET_HOME/extensions/chamfer_dist
python setup.py install --user
cd $GRNET_HOME/extensions/cubic_feature_sampling
python setup.py install --user
cd $GRNET_HOME/extensions/gridding
python setup.py install --user
cd $GRNET_HOME/extensions/gridding_loss
python setup.py install --user

运行

python runner.py
python runner.py --test --weights=<path-to-weights-file>

尽情享用吧~