pytorch3dの動作確認です
環境
Ubuntu 18.04 LST
GPU RTX2080Ti
CUDA 10.1
miniconda 4.7.12
環境構築
https://github.com/facebookresearch/pytorch3d/blob/master/INSTALL.md
インストールガイドに沿って環境構築を行いました
git clone
git clone https://github.com/facebookresearch/pytorch3d.git
conda 環境作成
conda create -n pytorch3d python=3.6
conda activate pytorch3d
conda install -c pytorch pytorch torchvision cudatoolkit=10.0
conda install -c conda-forge -c fvcore fvcore
condaでcuda10.0環境を改めて構築する
デモ用環境の構築
pip install scikit-image matplotlib imageio
conda install jupyter
pip install black isort flake8
pytorch3d本体の環境構築
conda install pytorch3d -c pytorch3d
demo実行
notebookを起動させます
cd ./pytorch3d
jupyter-notebook
このnotebook自体では毎回pytorch3dをビルドするのでそこそこ実行時間がかかります
./docs/tutorials内にあるbundle_adjustment.ipynbをnotebookで順次実行していきます
こんな感じで推論されたカメラ角度が出力されるはずです