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()