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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_30638831/article/details/83148308

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

猜你喜欢

转载自blog.csdn.net/qq_30638831/article/details/83148308