论文链接地址:https://arxiv.org/abs/1807.10221
论文原项目开源地址:https://github.com/CSAILVision/unifiedparsing
环境准备
pytorch1.0,cuda10.0,python3.6.6,Ubuntu20.04
cuda安装教程参考: https://blog.csdn.net/weixin_49214724/article/details/117744166
pytorch1.0版本安装使用conda,参考链接 https://pytorch.org/get-started/previous-versions/
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
注意事项
源码是2018年的版本,里面的scipy.misc中imread和imresize已经被弃用
# imread替换使用imageio包, imresize使用Pillow包替换
# from scipy.misc import imread,imresize
from