Python实现对图像进行掩膜遮罩处理

﹏ヽ暗。殇╰゛Y 2022-11-21 09:50 325阅读 0赞

**图像掩模（image mask）**：用选定的图像、图形或物体，对待处理的图像（局部或全部）进行遮挡来控制图像处理的区域或处理过程。由于覆盖的特定图像或物体称为掩模(mask)，在做图像处理的时候，对图像进行遮罩的需求非常多，接下来就以下面这张猫狗的图片进行演示，我选择的是小猫的头像。  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70][]

首先看效果图：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 1][]

# ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 2][] #

# 导入所需的库 #

此次需要的库资源有`cv2`和`numpy`，可通过`pip install xxx`进行下载。

import cv2
    import numpy as np

# 创建掩膜图像 #

创建掩膜具体需要看图像的尺寸，根据图像的尺寸所需创建自己的掩膜，当然掩膜的想也可以自己选择。我这里创建的掩膜有方形掩膜和圆形掩膜。

## 方形掩膜 ##

掩膜坐标为\[10:170, 50:220\]。

# 创建掩膜
    mask = np.zeros([img.shape[0], img.shape[1]], dtype=np.uint8)
    mask[10:170, 50:220] = 255

## 圆形掩膜 ##

掩膜坐标：  
x = 140  
y = 100  
r = 80

# 创建掩膜
    x = 140
    y = 100
    r = 80
    mask = np.zeros(img.shape[:2], dtype=np.uint8)
    mask = cv2.circle(mask, (x, y), r, (255, 255, 255), -1)

# 掩膜与原始图像进行拼接 #

图片合并用到了 cv2.add，将掩膜与原始图像进行拼接合并。

image = cv2.add(img, np.zeros(np.shape(img), dtype=np.uint8), mask=mask)

# 展示图像 #

# 展示原图
    cv2.imshow("img", img)
    # 展示掩膜图片
    cv2.imshow("mask", mask)
    # 展示添加掩膜效果图片
    cv2.imshow("image", image)

# 效果展示 #

原始图像：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 3][]  
方形掩膜图像：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 4][]  
方形掩膜与原图合并图像：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 5][]  
圆形掩膜图像：  
![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 6][]  
圆形掩膜与原图合并图像：

# ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 7][] #

# 总结 #

遮挡掩膜的原理很简单，首先创建一个和图片同样大小的全黑图像，再把需要显示的区域像素改成白色，最后使用cv2.add叠加image和mask就可以实现图像的遮挡显示。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70]: /images/20221120/f7618422c80443dea8b90989f663f930.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 1]: /images/20221120/4c7848f69dbb47cc9fc929c0a0a9c36f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 2]: /images/20221120/2457e6441dba4f4b995c3be0b193894e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 3]: https://img-blog.csdnimg.cn/20201031135333792.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 4]: https://img-blog.csdnimg.cn/20201031135339909.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 5]: https://img-blog.csdnimg.cn/20201031135343318.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 6]: https://img-blog.csdnimg.cn/20201031135347802.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz_size_16_color_FFFFFF_t_70 7]: https://img-blog.csdnimg.cn/202010311353528.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxNDcwNjAzODIz,size_16,color_FFFFFF,t_70