版权声明:本文为博主原创文章,未经博主允许不得转载。 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()