DeepLab: Deep Labelling for Semantic Image Segmentation で Sematic Segmentation する、と偉そうなことが書いてありますが、デモを動かすところまでです、すいません。
もののけ姫の1シーンを読み込ませてみると、鳥と人と馬というなかなかありそうな判定で面白かったです。
Installation
- 基本は書いてある通りに実施する。
- ただし以下に注意に注意。
- installation の
python deeplab/model_test.py
まで動けば大丈夫
注意
-
pip install tensorflow
すると、現状だと tensorflow ver.2 が入りますが、前提が ver.1 なので、バージョン指定でインストールしましょう。ver.2 でやるとエラーが出ます。 -
pip install tensorflow-gpu
は gpu版が要らなければ入れなくて大丈夫です。逆に入れるとエラーが出て動かなくなります。よくわからなかったら、とりあえずスルーした方が無難かもしれません。 -
sudo apt-get install python-pil python-numpy
は、mac であれば Pillow と numpy が入っていれば大丈夫そう(と信じて実行したら大丈夫だったけで、嘘かもしれません)。確認方法は、pip list
て表示されれば大丈夫です。されなければpip install
してあげてください。
DeepLab Demo.ipynb を動かす
- git clone
url
- cd models
- jupyter notebook deeplab/deeplab_demo.ipynb
↑を実行すると、ブラウザで勝手にhttp://localhost:8888/notebooks/deeplab_demo.ipynb
にアクセスされるはず。 - run all しろと書いてあるが、run all すると import 不足のエラーが出るので足して(下参照)から実施する。
run all
は、メニューのCell
から選べます。 - 完了すると入力した画像に合わせて以下のような結果が表示される。
import の追加
Select a pretrained model
import tensorflow as tf
Run on sample images
from six.moves import urllib
from PIL import Image
from io import BytesIO