-
这个源码是pspnet-pytorch实现,可用于训练个性化模型
资源介绍
PSPnet:Pyramid Scene Parsing Network语义分割模型在Pytorch当中的实现
目录
性能情况
训练数据集
权值文件名称
测试数据集
输入图片大小
mIOU
VOC12+SBD
VOC-Val12
473x473
68.59
VOC12+SBD
VOC-Val12
473x473
81.44
所需环境
torch==1.2.0
注意事项
代码中的pspnet_mobilenetv2.pth和pspnet_resnet50.pth是基于VOC拓展数据集训练的。训练和预测时注意修改backbone。
文件下载
训练所需的pspnet_mobilenetv2.pth和pspnet_resnet50.pth可在百度网盘中下载。
链接: https://pan.baidu.com/s/1JX0BoAroPChBQrXYnybqzg 提取码: papc
VOC拓展数据集