图像分割将标注（label）的mask表示在图像image上

分割中将label和原图image表示在一起

import matplotlib.pyplot as plt

image = plt.imread('./image.png')
mask = plt.imread('./mask.png')
pred = plt.imread('./pred.png')

fig, ax = plt.subplots()
ax.imshow(image)
ax.imshow(mask, cmap='gray', alpha=0.5)
# 预测结果也表示出来
ax.imshow(pred, cmap='gray', alpha=0.5)

# fig.show()
fig.savefig('./overlapped.png')

原文链接：link