因為最近在做深度學習摳圖，正好要用到蒙版進行摳圖，所以我將摳圖代碼進行了封裝注釋，可以直接使用。可能走了彎路，若有高見請一定提出！

主要代碼

import cv2from PIL import Imageimport numpy as npclass 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’): ''' 將輸入的圖片經過蒙版轉化為透明圖構造函數(shù) :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

mask.jpg

output.png

到此這篇關于python利用蒙版摳圖(使用PIL.Image和cv2)輸出透明背景圖的文章就介紹到這了,更多相關python 輸出透明背景圖內容請搜索好吧啦網以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持好吧啦網！