ismaeIfm / image-segmenter-ios-deeplabv3

Ios integrated deeplab model Implementation of the Semantic Segmentation

Image Segmenter ios

Core ML Image Segmentation on ios

Examples

mlmodel

Core ML model file are converted from deeplab MobileNetv2

Example

Download checkpoint and uncompress it

wget http://download.tensorflow.org/models/deeplabv3_mnv2_pascal_trainval_2018_01_29.tar.gz
tar xvf deeplabv3_mnv2_pascal_trainval_2018_01_29.tar.gz

Then convert to mlmodel with tfcoreml

import tfcoreml as tf_converter 
tf_converter.convert(
    tf_model_path = 'deeplabv3_mnv2_pascal_trainval/frozen_inference_graph.pb', 
    mlmodel_path = 'Deeplab.mlmodel', 
    image_input_names=["ImageTensor:0"], 
    output_feature_names = ['ResizeBilinear_3:0'], 
    input_name_shape_dict = {'ImageTensor:0' : [1, 513, 513, 3]}, 
    use_coreml_3 = False) 

Usage

 let sourceImg = UIImage.init(named: "test.jpg")!
 let segmentImg = sourceImg.segmentation()