FCRN-DepthPredictionを使った画像の深度推定

ホーム
AI, Linux, python
FCRN-DepthPredictionを使った画像の深度推定

2020.07.1

FCRN-DepthPredictionを使った画像の深度推定

単眼カメラの画像に対してAIを使って深度の推定をする。
FCRN-DepthPredictionというオープンソースのソフトウェアを使う。

VirtualBoxを使ってUbuntu16.04LTSをインストール。
まっさらな環境から主に下記URLの通りの手順を踏んでインストール。

git clone https://github.com/iro-cp/FCRN-DepthPrediction.git
cd FCRN-DepthPrediction

virtualenvを使って、Pythonの仮想環境を使って作業を勧めていく。

sudo apt install virtualenv
virtualenv -p python 3.5 python 3.5
source 3.5/bin/activate

必要なパッケージのインストール

(python3.5)$ pip install tensorflow==1.15.0
(python3.5)$ pip install opencv-python
(python3.5)$ pip install matplotlib
(python3.5)$ pip install Pillow
(python3.5)$ pip install numpy
(python3.5)$ pip install argparse

深度計算の学習済みモデルを取得、展開

(python3.5)$ cd tensorflow/models
(python3.5)$ wget http://campar.in.tum.de/files/rupprecht/depthpred/NYU_FCRN-checkpoint.zip
(python3.5)$ unzip NYU_FCRN-checkpoint.zip

この時点でtensorflow/modelにあるファイル一覧

__init__.py
fcrn.py
NYU_FCRN.ckpt.index
__pycache__
fcrn.cpython-35.pyc
__init__.cpython-35.pyc
network.cpython-35.pyc
NYU_FCRN.ckpt.meta
NYU_FCRN.ckpt.data-00000-of-00001
network.py
NYU_FCRN-checkpoint.zip

tensorflowディレクトリに戻り、実行する

(python3.5)$ cd ..  ([クローンしたトップディレクトリ]/tensorflow に移動）
(python3.5)$ ls     (下記のファイル/フォルダがあるはず）
models                predict.py
(python3.5)$ python predict.py models/NYU_FCRN.ckpt [入力画像のファイルパス]

ここで、PythonのTK_interがないというエラーが出た。
これはpipではなく、aptを使ってインストールする。