pspnet-pytorch

pspnet-pytorch copied to clipboard

使用自己的数据集训练失败的问题？

1

train.py中把batch_size调小之后还是会出现“CUDA out of memory. Tried to allocate 4.16 GiB”的问题，请问是哪里需要调整呢？

mmammoth

你的代码只用voc2007做训练集吗

3

liu-ai-z

主干跟换后出现了如下报错： Epoch 1/130: 0%| | 0/1439 [00:00

chen-del

请问如何把resnet50更换为resnet152

7

请问如何把resnet50更换为resnet152

mark1229g

主干网络bankbone更换

1

更换为mobnetv3

chen-del

VOC2007 和 VOC2012

1

请问，我看原作者的pytorch仓库里写的数据集是 VOC2012_aug ，咱们这个是VOC2007_aug 请问内容上有区别吗？

zhijiejia

pspnet-resnet50.pth

4

请问下载的pspnet_resnet50.pth是作者您已经训练好的是吗？ 我在使用train.py 训练时，使用了Resnet50的预训练模型，是不是需要把加载pspnet_resnet50.pth的代码注释掉呢？

zhijiejia

测试图片mask全黑

9

你好，我用的自己的数据（两类）训练的模型，测试时用get_iou_prediction.py保存的图片mask为全黑的，我想问下这是什么情况呢，我需要修改代码的哪部分呢

mxy519

有关loss的问题

1

大佬！你好，想问下使用的混合损失函数都是线性组合吗？像dice和focal就是loss=dice loss +focal loss。自己不大确定想问问。谢谢！

Nine9844

SegmentationClass中有些图片全黑

1

博主，我想问一下在readme下载的voc扩展数据集中的SegmentationClass里的图片，有些边缘是255所以能看到一圈白边，有些却没有255的白色边缘。我想获得有白色边缘的分割图片，应该怎样做呢？

njust-lyy