Skip to content

Deeplab v3 segmentation model converted from TensorFlow to CoreML for on device object segmentation.

License

Notifications You must be signed in to change notification settings

IlyaKrotov/iOS-Object-Segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

iOS-Object-Segmentation

Deeplab v3 segmentation model converted from TensorFlow to CoreML for on-device object segmentation. Code for the model and pretrained weights were taken from https://github.com/sthalles/deeplab_v3. Also, this notebook provides a good example of how to convert tensorflow model to CoreML and then test it.

Usage

  1. Download model checkpoint from here https://www.dropbox.com/sh/s7sx69pqjhrk0s4/AACXWCRd9JJ0zvcvDES9G3sba?dl=0.
  2. Just run notebook and then you will have .coreml model and tensorflow frozen graph.

Result

Alt text Image is taken from https://instagram.com/asyaze.

I hope that I will add more interesting models and examples to this repo. If you want to add more examples of convertion or models, feel free to do it!

About

Deeplab v3 segmentation model converted from TensorFlow to CoreML for on device object segmentation.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published