车道图片与分割图片进行半透明叠加

最新推荐文章于 2023-02-25 11:28:06 发布

原创最新推荐文章于 2023-02-25 11:28:06 发布 · 719 阅读

3 ·

CC 4.0 BY-SA版权

车道线检测专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种将车道分割算法的结果与原始图像融合的方法，通过调整颜色权重使分割区域更直观地显示在图像上，便于评估分割算法的效果。

实现车道分割算法时需要将车道分割结果叠加到原图上，方便观察分割效果。以下图例来自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")

在这里插入图片描述