先放上代码
import glob
import cv2
# np.set_printoptions(threshold=np.inf)
label_path = r'oct/label224/*.png'
for i, label_path in enumerate(glob.glob(label_path)):
image_path = label_path.replace('label224', 'img224')
#读入图像
image = cv2.imread(image_path)
image = cv2.cvtColor(image,cv2.COLOR_BGR2RGB)
label = cv2.imread(label_path, flags=0)
print(image)
# 关键代码
image[label == 255] = [255,0,0]
print(image)
cv2.imwrite("out.png", image)
# print(i, label_path)
# print(label)
# print(label.shape)
图片数据结构如下图所示
标签中打标签部分为白色255,图片为灰度图,原图为三通道RGB格式
将255部分替换为红色[255,0,0]
将白色部分替换为红色放到原图,关键代码如下
image[label == 255] = [255,0,0]