作者:陈洪瀚 /洪瀚笔记知乎专栏
摘要:介绍了使用PyTorch和torchvision加载训练好的全卷积网络FCN或DeepLab模型,并对PASCAL VOC图像进行分割并显示结果。
网址:
github代码链接, 码云代码链接
陈洪瀚www.zhihu.com一. 准备实验数据
- 下载PASCAL VOC 2012数据集(训练和验证集)。Download the training/validation data (2GB tar file)
下载地址http://host.robots.ox.ac.uk/pascal/VOC/voc2012/index.html
2. 将数据解压 放到用户目录下 /home/<user>/VOC
二. 运行环境
Numpy, Torch, Torchvision, pillow, matplotlib
三. 程序
# file name: easy_voc_segmentation_test.py