**在 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")

处理前的图像

处理后的图像