python opencv 将图片非感兴趣区域黑白化
时间: 2023-10-06 18:12:22 浏览: 76
python 利用OpenCV 图像黑白化
4星 · 用户满意度95%
您可以通过以下步骤将非感兴趣区域变为黑白色:
1. 加载图像并创建一个掩码,用于标识感兴趣区域和非感兴趣区域。
```python
import cv2
img = cv2.imread('image.jpg')
mask = np.zeros(img.shape[:2], np.uint8)
roi_corners = np.array([[x1,y1],[x2,y2],[x3,y3],[x4,y4]], dtype=np.int32)
ignore_mask_color = (255,) * channel_count
cv2.fillPoly(mask, roi_corners, ignore_mask_color)
```
其中,`x1, y1, x2, y2, x3, y3, x4, y4` 是感兴趣区域的四个顶点坐标,`channel_count` 表示图像通道数量。
2. 将非感兴趣区域的像素值设为黑色。
```python
masked_img = cv2.bitwise_and(img, img, mask=mask)
gray_img = cv2.cvtColor(masked_img, cv2.COLOR_BGR2GRAY)
```
这里使用了 `cv2.bitwise_and()` 函数来将掩码应用到原始图像上,将非感兴趣区域的像素值设为零。然后使用 `cv2.cvtColor()` 函数将结果转换为灰度图像。
3. 将感兴趣区域的像素值设为白色。
```python
mask_inv = cv2.bitwise_not(mask)
white_img = np.full(img.shape, 255, dtype=np.uint8)
white_img = cv2.bitwise_and(white_img, white_img, mask=mask_inv)
result_img = cv2.bitwise_or(gray_img, white_img)
```
这里使用了 `cv2.bitwise_not()` 函数来反转掩码,将感兴趣区域的像素值设为白色。然后创建一个全白色的图像,使用 `cv2.bitwise_and()` 函数将掩码应用到图像上,将非感兴趣区域的像素值设为零。最后使用 `cv2.bitwise_or()` 函数将两个图像合并,得到最终结果。
完整代码如下:
```python
import cv2
import numpy as np
# Load image
img = cv2.imread('image.jpg')
# Create mask
mask = np.zeros(img.shape[:2], np.uint8)
roi_corners = np.array([[x1,y1],[x2,y2],[x3,y3],[x4,y4]], dtype=np.int32)
ignore_mask_color = (255,) * channel_count
cv2.fillPoly(mask, roi_corners, ignore_mask_color)
# Apply mask to image
masked_img = cv2.bitwise_and(img, img, mask=mask)
# Convert to grayscale
gray_img = cv2.cvtColor(masked_img, cv2.COLOR_BGR2GRAY)
# Make non-masked region white
mask_inv = cv2.bitwise_not(mask)
white_img = np.full(img.shape, 255, dtype=np.uint8)
white_img = cv2.bitwise_and(white_img, white_img, mask=mask_inv)
# Combine masked region and white region
result_img = cv2.bitwise_or(gray_img, white_img)
# Display result
cv2.imshow('Result', result_img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
阅读全文