VOC数据集颜色对应关系与代码

最新推荐文章于 2022-07-19 23:02:37 发布

原创最新推荐文章于 2022-07-19 23:02:37 发布 · 4.9k 阅读

28 ·

CC 4.0 BY-SA版权

tensorflow 同时被 2 个专栏收录

53 篇文章

订阅专栏

图像

21 篇文章

订阅专栏

本文深入探讨了VOC数据集中颜色与分类的关系，详细解释了一种生成颜色映射矩阵的算法，该矩阵用于视觉识别任务中不同类别颜色的标准化表示。

VOC颜色和分类的对于关系:

code:

def voc_colormap(N=256):
    def bitget(val, idx): return ((val & (1 << idx)) != 0)

    cmap = np.zeros((N, 3), dtype=np.uint8)
    for i in range(N):
        r = g = b = 0
        c = i
        for j in range(8):
            r |= (bitget(c, 0) << 7 - j)
            g |= (bitget(c, 1) << 7 - j)
            b |= (bitget(c, 2) << 7 - j)
            c >>= 3
        print([r, g, b])
        cmap[i, :] = [r, g, b]
    return cmap

VOC_COLORMAP = voc_colormap()