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)$

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

(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を使ってインストールする。

$ sudo apt-get install python3-tk

無事動作を確認。

下記URLを参考にさせていただきました。

Tensorflowの深度推定モデルを試してみる - Qiita

非常に助かりました。
ありがとうございます。

Posted in AI, Linux, python | No Comments

19

8月 19

awkを使って一行の中の複数の要素をマッチさせる方法

今年の末にyahooブログがサービスを終了するにともなって、そのブログをwordpressに移行したいという依頼があった。
その中で、インポート用のテキストファイルから画像ファイルのリスト化をする処理の部分でawkを使ったところ、一行に2つ以上のimgタグとsrcが存在した場合、上手く要素を取得できないという問題に行き当たった。

こういう状態↓

私は最初、キャプチャーの位置を()でグルーピングして\1,\2などで取得するという方法しかわからなかった。
しかし、これでは当然、一行に3つ以上の要素がある場合マッチできない。

結局的に以下のコードでうまく行った。

awk 'BEGIN{FS="src=";}
{
	for(i=2;i<=NF;++i){
		if(match($i,/"[^"]*"/)){
			print substr($i,RSTART+1,RLENGTH-2);
		}
	}
}' ./source.txt

フィールドセパレーターを"src="に設定してあげることで、一行はimgタグのsrcで分割される。
iの初期値が2から始まっているのは、awkのカウントは0オリジンではなく、1オリジンであり、なおかつ"src="で区切られたフィールドの2つ目からマッチの試行を行いたいため。
あとは行中のマッチ位置のインデックスをRSTARTとRLENGTHで取得し、substr関数で切り抜く。

Posted in awk, Linux, Unix, 未分類 | No Comments

29

7月 19

Seleniumを使おうとしてGoogleChromeが起動せずにハマった件

ウェブのフォーム入力テストの自動化をしたくて、Seleniumを導入しようとしたところ、下記のエラーが出て起動してくれない

環境:
Linux Ubuntu 18.04LTS
Python 3.7.3
pip 19.0.3
Google Chrome 75.0.3770.142
ChromeDriver 75.0.3770.140 (2d9f97485c7b07dc18a74666574f19176731995c-refs/branch-heads/3770@{#1155})

(unknown error: DevToolsActivePort file doesn't exist) (The process started from chrome location /usr/bin/google-chrome is no longer running, so ChromeDriver is assuming that Chrome has crashed.)

結論としては環境変数$DISPLAYの設定が漏れていたからだった。

echo 'export DISPLAY="${DISPLAY:-:0}"' >> ~/.zshrc

にて解決。

だめな人は単にこれでも動くかも

export DISPLAY=:0.0

解決までの道筋

はじめに下記のURLを参考にSeleniumをインストールした
Python + Selenium で Chrome の自動操作を一通り - Qiita

しかし動かない…Firefoxの方も試してみたけど動かない。

次に下記のURLを参考にスクリーンショットを取得してみる

UbuntuのCLI環境でGoogleChromeを使ってウェブサイトのスクリーンショットを取得する - Qiita

たしかにGoogleの検索画面のスクリーンショットが取得できた。
chromedriverは動いていることを確認

ということは、ブラウザの起動まわりで上手く行っていないということが伺える。

そこでchromeを単体で実行(/usr/bin/google-chrome)してみる。
(google-chrome:29464): Gtk-WARNING **: 17:41:15.291: cannot open display:

というエラーが出て起動しないことで、DISPLAY周りで異常が起こっていることを突き止める。

最終的に下記URLを参考に.zshrcでDISPLAY環境変数を設定できていなかったことに気づく。
ubuntu で root 時に cannot open display となる件 - たつこの部屋


#google.py
import time
from selenium import webdriver

driver = webdriver.Chrome()
driver.get('https://www.google.com/')
time.sleep(5)
search_box = driver.find_element_by_name("q")
search_box.send_keys('ChromeDriver')
search_box.submit()
time.sleep(5)
driver.quit()

参考文献など :
$DISPLAYの設定: ubuntu で root 時に cannot open display となる件 - たつこの部屋
環境変数$DISPLAYについて: DISPLAY - なんなんなん行く？
seleniumのインストール: UbuntuのCLI環境でGoogleChromeを使ってウェブサイトのスクリーンショットを取得する - Qiita

Posted in Linux, 未分類 | No Comments
Tags: assuming, Chrome, crashed, has, Selenium

WisdomTrees

Archive for the ‘Linux’ Category