Skip to content

Latest commit

 

History

History
45 lines (32 loc) · 903 Bytes

README.md

File metadata and controls

45 lines (32 loc) · 903 Bytes

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