**在 python 中保存帶有添加蒙版的圖像**

Question

我正在嘗試保存我在所有感興趣區域上添加了白色遮罩的圖像。 但出於某種原因，它沒有保存最終圖像，也沒有返回任何錯誤消息。 如何保存帶蒙版的圖像？

import cv2
import numpy as np


image = cv2.imread('C:/Users/Desktop/testim.png') 
gray_scale = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) 
# Set threshold level 

Dark = 10 

coords = np.column_stack(np.where(gray_scale < Dark)) 

print("xy:\n", coords)

mask = gray_scale < Dark

# Color the pixels in the mask 

image[mask] = (255, 255, 255) 

cv2.imshow('mask', image) 
cv2.waitKey()

#save new image with the added mask to directory 
if not cv2.imwrite(r'./mask.png', image):
     raise Exception("Could not write image")

Answer 1

我認為這與程序中的幾個錯別字有關。 修復它們后，一切都很好。

import cv2
import numpy as np


image = cv2.imread('/content/test.png') 
gray_scale = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) 
# Set threshold level 

Dark = 10 

coords = np.column_stack(np.where(gray_scale < Dark)) 

# print("xy:\n", coords)

mask = gray_scale < Dark

# Color the pixels in the mask 

image[mask] = (255, 255, 255) 
# cv2.imshow('mask', image) 
cv2.waitKey()

if not cv2.imwrite(r'./mask.png', image):
     raise Exception("Could not write image")

處理前的圖像

處理后的圖像