实现车道分割算法时需要将车道分割结果叠加到原图上,方便观察分割效果。以下图例来自bdd100k数据集
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
color_label_path = "color_label.png"
img_path = "img.jpg"
img = Image.open(img_path)
color_label = Image.open(color_label_path)
img = np.array(img, dtype="float64")
color_label = np.array(color_label, dtype="float64")
print(color_label.shape)
label_channels_sum = color_label.sum(2)
index = (label_channels_sum != 0)
img[index, :] += color_label[index, :]
img[index, :] /= 2
img /= 255.0
plt.imshow(img)
plt.savefig("./blend_img.png")