基于pytorch的UNet_demo实现及训练自己的数据集.docx

　　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-mang-unet-ask 文件夹下的一张包含有人的图像，Output.jpg 是输出。 若报错：No module named ‘ xxx ’ 如下：ModuleNotFoundError: No module named 'torchsummary'解决方法：pip 安装相应模块即可。(2) UNet-Pytorch 训练数据集第一步：下载数据集百度网盘：http://pan.baidu.com/s/1dE14537密码： ndg8