
作者:陳洪瀚 /洪瀚筆記知乎專欄
摘要:介紹了使用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