Home

Awesome

mobile_phone_human_matting

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.

requirements

Usage

train

you need to prepare dataset and run ./train.sh.

test

use pre_trained model ./pre_train/erd_seg_matting.

test camera, you need a camera, run ./camera.sh. test image, run ./test.sh.

Speed Analysis

Platform : ncnn.

(use this tools convert the pytorch model to ncnn.)

Mobile phone: Samsung Galaxy S8+(cpu).

model size (M)time(ms)
erd_seg_matting3.4~40

Demo video on my iphone 6 (baiduyun)

References

paper

repos