cd data
wget https://www.dropbox.com/s/35momrh68zuhkei/Figaro1k.zip
unzip Figaro1k.zip
cd checkpoints
wget https://github.com/YBIGTA/pytorch-hair-segmentation/blob/master/models/pspnet_resnet101_sgd_lr_0.002_epoch_100_test_iou_0.918.pth
到src目录下,进行判断
python validation.py
- 网络加速 (目前是CPU 0.67fps)
- 部署到视频 (实时渲染)
- 尝试各种网络
reference pytorch-hair-segmentation