daniel985 / MobileNet4Segmentation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MobileNet4Segmentation

MobileNet for foreground-background segmentation

1.You need to get tensorflow models:

git clone https://github.com/tensorflow/models ~/tensorflow_models

2.Download pretrained MobileNet model from here:

https://github.com/tensorflow/models/blob/master/slim/nets/mobilenet_v1.md

3.Prepare 'RGBA' image dataset for train:

'A' is the mask for foreground-background

4.Use 'RGB' image for test

5.segMnet_ASPP.py provide ASPP module which from deeplab3:

https://arxiv.org/pdf/1706.05587.pdf

6.test_with_crf.py provide CRF module

About


Languages

Language:Python 100.0%