python利用蒙版抠图(使用PIL.Image和cv2)输出透明背景图

àì夳堔傛蜴生んèń 2022-12-20 05:49 200阅读 0赞

更多编程教程请到：[菜鸟教程][Link 1] https://www.piaodoo.com/

因为最近在做深度学习抠图，正好要用到蒙版进行抠图，所以我将抠图代码进行了封装注释，可以直接使用。可能走了弯路，若有高见请一定提出！

主要代码

import cv2
    from PIL import Image
    import numpy as np

class UnsupportedFormat(Exception):  
def **init**(self, input\_type):  
self.t = input\_type

def **str**(self):  
return “不支持’\{\}'模式的转换，请使用为图片地址(path)、PIL.Image(pil)或OpenCV(cv2)模式”.format(self.t)

class MatteMatting():  
def **init**(self, original\_graph, mask\_graph, input\_type=‘path’):  
“”"  
将输入的图片经过蒙版转化为透明图构造函数  
:param original\_graph:输入的图片地址、PIL格式、CV2格式  
:param mask\_graph:蒙版的图片地址、PIL格式、CV2格式  
:param input\_type:输入的类型，有path：图片地址、pil：pil类型、cv2类型  
“”"  
if input\_type == ‘path’:  
self.img1 = cv2.imread(original\_graph)  
self.img2 = cv2.imread(mask\_graph)  
elif input\_type == ‘pil’:  
self.img1 = self.\_\_image\_to\_opencv(original\_graph)  
self.img2 = self.\_\_image\_to\_opencv(mask\_graph)  
elif input\_type == ‘cv2’:  
self.img1 = original\_graph  
self.img2 = mask\_graph  
else:  
raise UnsupportedFormat(input\_type)

@staticmethod  
def \_\_transparent\_back(img):  
“”"  
:param img: 传入图片地址  
:return: 返回替换白色后的透明图  
“”"  
img = img.convert(‘RGBA’)  
L, H = img.size  
color\_0 = (255, 255, 255, 255) \# 要替换的颜色  
for h in range(H):  
for l in range(L):  
dot = (l, h)  
color\_1 = img.getpixel(dot)  
if color\_1 == color\_0:  
color\_1 = color\_1\[:-1\] + (0,)  
img.putpixel(dot, color\_1)  
return img

def save\_image(self, path, mask\_flip=False):  
“”"  
用于保存透明图  
:param path: 保存位置  
:param mask\_flip: 蒙版翻转，将蒙版的黑白颜色翻转;True翻转;False不使用翻转  
“”"  
if mask\_flip:  
img2 = cv2.bitwise\_not(self.img2) \# 黑白翻转  
image = cv2.add(self.img1, img2)  
image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR\_BGR2RGB)) \# OpenCV转换成PIL.Image格式  
img = self.\_\_transparent\_back(image)  
img.save(path)

@staticmethod  
def \_\_image\_to\_opencv(image):  
“”"  
PIL.Image转换成OpenCV格式  
“”"  
img = cv2.cvtColor(np.asarray(image), cv2.COLOR\_RGB2BGR)  
return img

使用示例

mm = MatteMatting("input.jpg", "mask.jpg")
    mm.save_image("output.png", mask_flip=True) # mask_flip是指蒙版翻转，即把白色的变成黑色的，黑色的变成白色的

效果展示

`input.jpg`

![Image 1][]

`mask.jpg`

![Image 1][]

`output.png`

![Image 1][]

到此这篇关于python利用蒙版抠图(使用PIL.Image和cv2)输出透明背景图的文章就介绍到这了,更多相关python 输出透明背景图内容请搜索菜鸟教程www.piaodoo.com以前的文章或继续浏览下面的相关文章希望大家以后多多支持菜鸟教程www.piaodoo.com！

[Link 1]: https://www.piaodoo.com/
[Image 1]: