python抠透明图_python利用蒙版抠图(使用PIL.Image和cv2)输出透明背景图
因為最近在做深度學習摳圖，正好要用到蒙版進行摳圖，所以我將摳圖代碼進行了封裝注釋，可以直接使用。可能走了彎路，若有高見請一定提出！
主要代碼
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 "不支持'{}'模式的轉(zhuǎn)換，請使用為圖片地址(path)、PIL.Image(pil)或OpenCV(cv2)模式".format(self.t)
class MatteMatting():
def __init__(self, original_graph, mask_graph, input_type='path'):
"""
將輸入的圖片經(jīng)過蒙版轉(zhuǎn)化為透明圖構造函數(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: 蒙版翻轉(zhuǎn)，將蒙版的黑白顏色翻轉(zhuǎn);True翻轉(zhuǎn);False不使用翻轉(zhuǎn)
"""
if mask_flip:
img2 = cv2.bitwise_not(self.img2) # 黑白翻轉(zhuǎn)
image = cv2.add(self.img1, img2)
image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)) # OpenCV轉(zhuǎn)換成PIL.Image格式
img = self.__transparent_back(image)
img.save(path)
@staticmethod
def __image_to_opencv(image):
"""
PIL.Image轉(zhuǎn)換成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是指蒙版翻轉(zhuǎn)，即把白色的變成黑色的，黑色的變成白色的
效果展示
input.jpg
mask.jpg
output.png
到此這篇關于python利用蒙版摳圖(使用PIL.Image和cv2)輸出透明背景圖的文章就介紹到這了,更多相關python 輸出透明背景圖內(nèi)容請搜索我們以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持我們！
本文標題: python利用蒙版摳圖(使用PIL.Image和cv2)輸出透明背景圖
本文地址: http://www.cppcns.com/jiaoben/python/330247.html
總結
以上是生活随笔為你收集整理的python抠透明图_python利用蒙版抠图(使用PIL.Image和cv2)输出透明背景图的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： python重新加载模块_jupyter
- 下一篇： python 接口测试 url_Pyth