DUTS显著性检测数据集mask格式为RGB的问题及解决脚本

最新推荐文章于 2025-04-12 22:56:46 发布

塔克拉玛干沙漠的卖水小孩

最新推荐文章于 2025-04-12 22:56:46 发布

阅读量1.1k

点赞数 3

分类专栏：其他文章标签：计算机视觉

本文链接：https://blog.csdn.net/qq_49230659/article/details/121175221

版权

其他专栏收录该内容

1 篇文章

订阅专栏

该博客主要讲述了DUTS数据集中存在的一些问题，即部分灰度图片以RGB三通道格式存储，可能导致处理时出现错误。作者提供了一个Python脚本，用于将这些24位的RGB灰度图片转换为8位的L格式，以解决潜在的bug。受影响的图片包括DUTS-TR和DUTS-TE数据集中的特定文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

DUTS数据集有几个GT label格式是RGB，虽然是灰度图片，但是却是以三通道的格式储存的，在用的时候可能会引起一些bug，因此我们最好把数据集格式优化一下，我发现有这么几张图片有问题，写在下面的脚本里了。

from PIL import Image

"""DUTS-TR数据集的ILSVRC2014_train_00023530/n01532829_13482/n04442312_17818是一张黑白图片（GT），但是却有3个通道，所以会报错;
DUTS-TE数据集的ILSVRC2012_test_00036002.png, sun_bcogaqperiljqupq.png也是一样的情况
这个脚本把这个图像24位变成8位（RGB格式转换成L格式）"""

img_path = 'XXX（你的路径）/DUTS-TR-Mask/ILSVRC2014_train_00023530.png'
img = Image.open(img_path)
im = img.convert('L')

im.save(img_path)