颜色标签不一致这么解决?
收藏
我用官方的deeplab V3+教程测试,结果输出图的颜色标签与训练集的颜色标签不一致,请问各位大佬,这是什么情况了(无语,上传不了图片)
0
收藏
请登录后评论
颜色标签什么意思
是预测框的线条颜色问题吗
不是预测框,是输出图的颜色,真是无语,发不了图片。这是实验链接https://aistudio.baidu.com/aistudio/projectdetail/743111,举例来说,结果图里面道路是黑色的,车是紫色的,而数据集里面,车是蓝色,道路是但粉红,所以我就很好奇,为什么回这样
我这里的颜色标签是这样的:在cityscapes数据集里面,一个物体类别对应一种颜色标签,如车是蓝色,路是粉红,我的问题是,输出图颜色与训练集的颜色对不上,我的输出图里面,车是紫色,道路则是黑色的,所以想请教各位大佬,这是什么情况,这是我的实验链接https://aistudio.baidu.com/aistudio/projectdetail/743111
你的项目没有公开,另外我记得有个文件是可以专门配置颜色显示的,文件或函数名字里好像有个map
可能是映射关系搞错了,找到那个显示颜色的部分回溯下应该能找到
显示颜色部分,好像只有图片,没有颜色标签设置,,
教程里面好像没有呀,大神
看看这个issue?
https://github.com/PaddlePaddle/PaddleSeg/issues/82
感谢 大佬的指点
大佬,我看教程里面的color-map好像也没什么问题呀,这是在iferance.tools.visualize.py的代码
import cv2
import sys
# ColorMap for visualization more clearly
color_map = [[128, 64, 128], [244, 35, 231], [69, 69, 69], [102, 102, 156],
[190, 153, 153], [153, 153, 153], [250, 170, 29], [219, 219, 0],
[106, 142, 35], [152, 250, 152], [69, 129, 180], [219, 19, 60],
[255, 0, 0], [0, 0, 142], [0, 0, 69], [0, 60, 100], [0, 79, 100],
[0, 0, 230], [119, 10, 32]]
im = cv2.imread(sys.argv[1])
# Please note (224, 224) just for daheng model
print("visualizing...")
for i in range(0, 224):
for j in range(0, 224):
im[i, j] = color_map[im[i, j, 0]]
cv2.imwrite(sys.argv[1], im)
print("visualizing done!")
我看这里面的颜色标签好像也没什么问题呀(与数据集的标签差不多),因为我也不知道具体的标签怎么排序,请问可以指教一下吗
这是我按照教程仿真的实验https://aistudio.baidu.com/aistudio/projectdetail/192369
我pedseg里面找到您说的vis.py中的get_color_map函数了
def get_color_map(num_classes):
""" Returns the color map for visualizing the segmentation mask,
which can support arbitrary number of classes.
Args:
num_classes: Number of classes
Returns:
The color map
"""
#color_map = num_classes * 3 * [0]
color_map = num_classes * [[0, 0, 0]]
for i in range(0, num_classes):
j = 0
color_map[i] = [0, 0, 0]
lab = i
while lab:
color_map[i][0] |= (((lab >> 0) & 1) << (7 - j))
color_map[i][1] |= (((lab >> 1) & 1) << (7 - j))
color_map[i][2] |= (((lab >> 2) & 1) << (7 - j))
j += 1
lab >>= 3
return color_map
请问应该怎么改,才能让输出图的颜色标签与训练集里面的颜色标签保持一致了,这是我的仿真实验https://aistudio.baidu.com/aistudio/projectdetail/192369
突然想到个问题,难道是因为类别数变了?我觉得可以去issue上问问。项目我得找时间看下,因为其实我也遇到了颜色的问题,不过更复杂还没研究清楚,我之前以为是类别数的问题……
好的,感谢指点
路过 前来学习
您好,我也遇到了同样的问题,请问您现在解决了吗
visualize_segmentation(image,
result,
weight=0.6,
save_dir='./',
color)
要设置这个函数的color参数,传递格式为BGR,list;
例如二分类是 [255,255,255,0,0,0]
他后面写了函数按照没类别三个自己获取颜色编号