/pytorch_PortraitNet

pytorch_PortraitNet

Primary LanguagePython

PortraitNet in PyTorch

This repository contains the PortraitNet (reproduced by PyTorch) for "PortraitNet: Real-time portrait segmentation network for mobile device." which modified from MobileNetV2_unet.

Overview

PortraitNet

Speed and Model size analysis

PortraitNet

Contact

pfw813@gmail.com

Any discussions or concerns are welcomed!

Citation

If you find this paper useful in your research, please consider citing:

@article{
  title={ PortraitNet: Real-time Portrait Segmentation Network for Mobile Device},
  author={Song-Hai Zhang, Xin Dong, Jia Li, Ruilong Li, Yong-Liang Yang},
  journal={Computers & Graphics},
  year={2019}
}

Acknowledgement

  1. Akira Sosa whose blog ''Real-Time deep learning in mobile application'' is a very excellent tutorial for practical application.