Hyojin Park, Lars Lowe Sjösund, YoungJoon Yoo, Nojun Kwak . "ExtremeC3Net: Extreme Lightweight Portrait Segmentation Networks using Advanced C3-modules" arXiv preprint [arxiv]
- python 3
- pytorch >= 0.4.0
- torchvision==0.2.1
- opencv-python==3.4.2.17
- numpy
- tensorboardX
- visdom
- Train
python mainwithEdge.py
- Stage1_ExtremeC3NetV2
- stage1 train CoarseNet
- stage2 train Full model
- ExtremeC3Net_small
- not use FineNet
- demo with video
# genotype from search results
python demo_video.py