train on dataset iDRiD,iou is very small

Qsingle / LearnablePromptSAM

Try to use the SAM-ViT as the backbone to create the learnable prompt for semantic segmentation

Apache License 2.0

84 stars 14 forks source link

Open liyan20180808 opened 2 months ago

liyan20180808 commented 2 months ago

您好，我在iDRiD数据集，使用代码进行复现，训练过程中，iou很小且一直下降，使用权重进行预测，出来的图是全黑的，完全没有预测正确，请问这是什么问题呢？谢谢！可以提供一下训练这个数据集的相关参数吗？

Qsingle commented 2 months ago

您好，我在iDRiD数据集，使用代码进行复现，训练过程中，iou很小且一直下降，使用权重进行预测，出来的图是全黑的，完全没有预测正确，请问这是什么问题呢？谢谢！可以提供一下训练这个数据集的相关参数吗？

参数与论文中一致，可能是由于标签转换的时候出现了问题，IDRiD的标签需要进行转换才能进行使用,可以参考 #6