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%