DeepLabv3+をipythonで試してみましょう

こんにちは。新年度になりましたね。

セマンティックセグメンテーションを使って猫を切り抜いて雑コラを作りたいと思っていた中、DeepLabv3+が少し話題になっていたので触ってみました。
https://research.googleblog.com/2018/03/semantic-image-segmentation-with.html
https://github.com/tensorflow/models/tree/master/research/deeplab

macへの導入に苦戦してしまってなんとなくの履歴しか残っていなかったのですがメモ程度に残しておきます。DeepLab、TensorFlow共に3月末時点のバージョンです。

もともとTensorFlowは試したことあって入っていたのですが、改めてバージョンアップしてからDeepLabを入れました。

スクリーンショット 2018-03-29 17.01.41

local_test.shまではここを見てなんとか進んだ(EvalやVisualizingに結構時間かかった…)んですが、jupyterが全然動かない…

周辺ライブラリとか本体を入れなおしたり試行錯誤してipythonで動きました。

うまく実行できるとブラウザでこういった画面が起動します。deeplab_demo.ipynbがサンプルなのでこれをクリック。

スクリーンショット 2018-04-02 13.14.45

次にこういった画面になり、上のブロックから順番に選択して(選択したものに枠が付きます)Runのボタンを押して行きます。自分の環境では一番上のブロックでPILとTensorFlowのインポートエラーが起きて、インストールし直しました。

スクリーンショット 2018-04-02 13.21.20

一番下のブロックまで実行して成功するとサンプルイメージのセマンティックセグメンテーションが行なわれます。自転車に乗っている子どものシルエットが出力されました。

スクリーンショット 2018-04-02 13.29.06

ちなみに最終ブロックのIMAGE_URLを書き換えるとサンプル以外の画像にも適用が可能です。

スクリーンショット 2018-04-02 13.32.52

ももさんもちゃんと猫として分離されました!

スクリーンショット 2018-04-02 13.36.44

これからPILで画像出力されている情報を数値で取得する作業を始めたいと思います。
果たしてできるのかはわからないですが!
ピクセル情報を取得できれば雑コラにも一歩近づけますね(゜-゜)

●この記事を書いた人