akirasosa/mobile-semantic-segmentation
A semantic segmentation model using MobileNetV2-inspired U-Net architecture for real-time mobile hair segmentation.

Velocity · 7d
+0.2
★ / day
Trend
→steady
star history
This project implements a deep learning model for real-time semantic segmentation optimized for mobile deployment. The architecture combines MobileNetV2 depthwise separable convolutions with a U-Net encoder-decoder structure to achieve 0.89 IoU on face/hair segmentation. It includes a training pipeline with dice coefficient loss, supports both PyTorch and TensorFlow, and exports to CoreML for iOS deployment and TensorFlow Lite for Android.