pyTorch 基於以resnet50為backbone的PSPNet 訓練VOC2012數據集

代碼鏈接：https://github.com/ggyyzm/pytorch_segmentation

 

使用PSPNet作為主干分類網絡

1.將VOC2012數據集下載並解壓到data/VOCtrainval_11-May-2012中

 

2.出現以下錯誤

 

3.經過調試發現是某一步中運行時間太長。將項目放到服務器上跑沒有此問題。

4.出現ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,512,1,1])

根據https://www.cnblogs.com/zmbreathing/p/pyTorch_BN_error.html解決問題，成功跑通。跑完結果如下

 acc達到93%，mIOU達到83%