Monodepth2を動かすメモ(仮)
2240 ワード
概要
- CUDA10.0のDocker imageで環境構築
- PyTorchのインストール
- Monodepth2を動かす
結論
- すぐ動かせるので簡単に試せる!すごい!!
てst
- 「Digging Into Self-Supervised Monocular Depth Estimation」のソースコードがGitHubで公開されている
- testのサンプルを動かすところまで
環境構築
- Anacondaは使わない
docker pull nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
docker run --runtime=nvidia -it --rm -e DISPLAY=$DISPLAY -v /tmp/.X11-unix/:/tmp/.X11-unix -e QT_X11_NO_MITSHM=1 -v /data:/data -w=/data nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
cd /data && git clone https://github.com/nianticlabs/monodepth2.git
関連インストール
- 流用しているので不要なライブラリも含む
apt update
apt-get install -yq --no-install-recommends --no-upgrade software-properties-common curl python3-dev python3-tk locales libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler libboost-all-dev libgflags-dev libgoogle-glog-dev liblmdb-dev opencl-headers ocl-icd-opencl-dev libviennacl-dev libopenexr-dev libsm6 libxext6 libxrender-dev
curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
python3 get-pip.py --force-reinstall
pip install --no-cache-dir Cython numpy protobuf openexr torch torchvision tensorboardX matplotlib opencv-python
実行
cd /data/monodepth2/
python3 test_simple.py --image_path assets/test_image.jpg --model_name mono+stereo_640x192
まとめ
- 親切な設計とREADMEはありがたい…
Author And Source
この問題について(Monodepth2を動かすメモ(仮)), 我々は、より多くの情報をここで見つけました https://qiita.com/nakasuke_/items/5675839c8ed469c4f688著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Content is automatically searched and collected through network algorithms . If there is a violation . Please contact us . We will adjust (correct author information ,or delete content ) as soon as possible .