如何在python中顯示地面實況圖像分割掩碼圖像？

Question

我試過這段代碼

from skimage import io
temp = io.imread(mask_input_path)
plt.imshow(temp)

這會顯示一個正常的圖像，因此輸出是黑色的。

Answer 1

就像普通圖像一樣。 如果您的整個蒙版是黑色的，則意味着您的圖像中不存在所需的對象。

但是要僅選擇蒙版區域，您需要在代碼上多加兩行

import matplotlib.pyplot as plt
input_img = plt.imread('img.jpg')
mask_img  = plt.imread('mask.jpg')

# select only masked area below
masked = input_img.copy()
masked[mask_img == 0 ] = 0

fig, axes = plt.subplots(1, 3, figsize=(16, 12))
ax = axes.flatten()

ax[0].imshow(input_img, cmap="gray")
ax[0].set_axis_off()
ax[0].set_title("Original Imput Image", fontsize=12)

ax[1].imshow(mask_img, cmap="gray")
ax[1].set_axis_off()
ax[1].set_title("Mask", fontsize=12)

ax[2].imshow(masked, cmap="gray")
ax[2].set_axis_off()
ax[2].set_title("Masked", fontsize=12)

plt.show()

實際使用

masked[mask_img < 30 ] = 0

給出稍微好一點的結果，因為在我的情況下，掩碼值並不完全為零