代码链接: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解决问题,成功跑通。
原文:https://www.cnblogs.com/zmbreathing/p/pyTorch_VOC.html