← all repositories

akirasosa/mobile-semantic-segmentation

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

722 stars Python Computer Vision
mobile-semantic-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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.