This project provides training and testing codes to build real_time human mattig on mobile phone only based CPU. Anather repo (Fast_Portrait_Segmentation) show more details about whence the project.
- python3.5 / 3.6
- pytorch 0.4/0.4.1
- opencv-python
you need to prepare dataset and run ./train.sh.
use pre_trained model ./pre_train/erd_seg_matting.
test camera, you need a camera, run ./camera.sh.
test image, run ./test.sh.
Platform : ncnn.
(use this tools convert the pytorch model to ncnn.)
Mobile phone: Samsung Galaxy S8+(cpu).
| model size (M) | time(ms) | |
|---|---|---|
| erd_seg_matting | 3.4 | ~40 |
Demo video on my iphone 6 (baiduyun)