Skip to content

afsaredrisy/Segmentation-CoreMl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

ios-Image-Segmentation

An application of image segmentation on iOS with U-Net. The format of the model is converted from PyTorch to Core ML.

Hair segmentation

Segment the hair from the image with U-Net.

Original image Segmented image
Original image Segmented image

TODO

  1. Create a repo of PyTorch model.

  2. Dye the hair with OpenCV.

  3. Face segmentation.

  4. Backgound segmentation.

Reference

[1] MobileNetV2

[2] U-Net

Releases

No releases published

Packages

No packages published

Languages