1. 根据官方 github 改编的一个人像分割例子
(1) UNet-Pytorch Demo
网址：https://github.com/leijue222/portrait-matting-unet-ask(分割人像)
https://cloud.tencent.com/developer/article/1658426
第一步：
$ git clone https://github.com/leijue222/portrait-matting-unet-flask.git
注：clone 下来已经包含 MODEL.pth
第二步：
$ cd portrait-matting-unet-flask/
$ python predict.py -i image.jpg -o output.jpg
注：image.jpg 是自己放在 portrait-mang-unet-ask 文件夹下的一张包含有人的
图像，Output.jpg 是输出。
若报错：No module named ‘ xxx ’ 如下：
ModuleNotFoundError: No module named 'torchsummary'
解决方法：pip 安装相应模块即可。
(2) UNet-Pytorch 训练数据集
第一步：下载数据集
百度网盘：http://pan.baidu.com/s/1dE14537
密码： ndg8
评论1