Skip to content

daniel985/MobileNet4Segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Releases

No releases published

Packages

 
 
 

Languages