search

PaddlePaddle / PaddleRS

Awesome Remote Sensing Toolkit based on PaddlePaddle.
Apache License 2.0
365 stars 88 forks source link

关于使用paddleRS的地物分类模型出现的问题 #187

Closed cctx1 closed 3 months ago

cctx1 commented 3 months ago

使用飞桨提供的DeepLab V3+模型对WHDLD数据集（数据集为jpg格式，label为png格式）进行地物分类训练，报错，报错结果如图1所示（ps：使用官方demo中的数据集正常运行，是数据的原因吗？），训练代码如图2所示。 图1： image 图2： image

欢迎您的提问。辛苦您提供以下信息，以方便我们快速定位和解决问题： PaddleRS版本：PaddleRS-1.0.0 PaddlePaddle版本：PaddlePaddle 2.4.0 操作系统信息：Windows Python版本号：Python3.7 CUDA/cuDNN版本：CUDA11.6/cuDNN 8.4.0

Bobholamovic commented 3 months ago

看起来是类别设置不正确或者数据集非标准格式，这个数据集存在标签6，但是模型输出只有6类。

cctx1 commented 3 months ago

我按照飞桨官方给的数据集类别来分的6类(如图1），划分数据集的完整代码如图2所示，生成的labels.txt里面也确实有6类，貌似没有错，但是为什么不行呢？ 图1： Q_5EXV7_UAY~DB 1B~3RUQU 图2： QQ截图20240309090435

Bobholamovic commented 3 months ago

你好，图一的数据集是社区的爱好者上传到AI Studio平台的，并不是PaddleRS官方提供的，因此我不能确定其中的数据格式一定满足PaddleRS要求～即使labels.txt中的类别符合要求，对于分割任务而言，标注的图像文件中也不能出现0-5以外的ID（当前很可能就是这种情况导致出错）。建议你可以检查数据集的标签图像中是否存在非法的数值～

cctx1 commented 3 months ago

确实是数据集label的问题，感谢开发者的回复~