使用python将EISeg分割生成的24位深的彩图转换为8位深的彩图

将EISeg分割生成的24位深的彩图转换为8位深的彩图

在训练语义分割时，需要有原图和相应的分割图，分割图一般有两种方式

一是用labelme进行打标签，然后根据json文件进行批量分割，这样的图片一般都是8位深的伪彩图，可以直接送到模型训练。

而第二种是使用EISeg生成的图片进行模型训练，然而EISeg生成的是24位深的伪彩图，如下图（肉眼看是没有任何区别的，但就是不能送到模型中训练），需要进行转换，转换为8位深的伪彩图。

下面展示将24位深的伪彩图转换为8位的伪彩图。

#img24to8
import os
import cv2
import numpy as np
from labelme.utils import lblsave
from skimage.color import hsv2rgb, rgb2hsv


def label_colormap(n_label=256, value=None):
    def bitget(byteval, idx):
        return (byteval & (1 << idx)) != 0

    cmap = np.zeros((n_label, 3), dtype=np.uint8)
    for i in range(0, n_label):
        id = i
        r, g, b = 0, 0, 0
        for j in range(0, 8):
            r = np.bitwise_or(r, (bitget(id, 0) << 7 - j))
            g = np.bitwise_or(g, (bitget(id, 1) << 7 - j))
            b = np.bitwise_or(b, (bitget(id, 2) << 7 - j))
            id = id >> 3
        cmap[i, 0] = r
        cmap[i, 1] = g
        cmap[i, 2] = b

    if value is not None:
        hsv = rgb2hsv(cmap.reshape(1, -1, 3))
        if isinstance(value, float):
            hsv[:, 1:, 2] = hsv[:, 1:, 2].astype(float) * value
        else:
            assert isinstance(value, int)
            hsv[:, 1:, 2] = value
        cmap = hsv2rgb(hsv).reshape(-1, 3)
    return cmap


colormap = label_colormap()

# 读入图片并将opencv的BGR转换为RGB格式
img = cv2.imread(r"C:\Users\User\Music\Desktop\打标签\1\Late_Blight_100_pseudo.png")
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

# 将24位深图片中的[r,g,b]对应到colormap反求出label
lbls = np.zeros(shape=[img.shape[0], img.shape[1]], dtype=np.int64)
len_colormap = len(colormap)
indexes = np.nonzero(img)

for i, j in zip(indexes[0], indexes[1]):
    for k in range(len_colormap):
        if all(img[i, j, :3] == colormap[k]):
            lbls[i, j] = k
            break


# 将label转换成8位
lblsave(os.path.join(os.getcwd(), r'C:\Users\User\Music\Desktop\1\Late_Blight_100.png'), lbls)

这样就可以将一张24位深的伪彩图转换为8位的伪彩图。

如果想要批量生成的，可以自行编写，也可以直接下载。
批量转换