pyTorch训练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解决问题，成功跑通。

原文：https://www.cnblogs.com/zmbreathing/p/pyTorch_VOC.html