Deeplabでリアルタイムにセグメンテーション

前回の続き

WebCamでリアルタイムに動かしたい！

ダウンロード

Real-time semantic image segmentation with DeepLab in Tensorflow

ページ下部の部分「deeplab_demo_webcam_v2.py」を使わさせていただきました。リンクからダウンロードしてtensorflow/models/research/deeplabに配置する。

モデルダウンロードのところにmobilenetv2のモデル追加

## Select and download models
_MODEL_URLS = {
   'mobilenetv2_coco_voctrainaug':'http://download.tensorflow.org/models/deeplabv3_mnv2_pascal_train_aug_2018_01_29.tar.gz',
   'mobilenetv2_coco_voctrainval':'http://download.tensorflow.org/models/deeplabv3_mnv2_pascal_trainval_2018_01_29.tar.gz',
   'xception_coco_voctrainaug': 'http://download.tensorflow.org/models/deeplabv3_pascal_train_aug_2018_01_04.tar.gz',
   'xception_coco_voctrainval': 'http://download.tensorflow.org/models/deeplabv3_pascal_trainval_2018_01_04.tar.gz',
}

_TARBALL_NAME = 'deeplab_model.tar.gz'

## fast
model_url = _MODEL_URLS['mobilenetv2_coco_voctrainaug']
## accurate
#model_url = _MODEL_URLS['xception_coco_voctrainaug']

opencvの追加
anacondaのtf-gpuにopencvを検索してインストール

anacondaのtf-gpu→OpenTerminalからdeeplab_demo_webcam_v2を起動

$ cd tensorflow\models\research\deeplab
$ python deeplab_demo_webcam.py

動く。
最高