У меня есть изображения знаков (знаков произвольной формы) с белым фоном, и я хочу получить изображение знака с прозрачным фоном.
Мне удалось создать маску и применить ее к изображению, и подумал, что сделать маску прозрачной можно. Я много искал здесь и в других местах, но мне ничего особо не помогло.
import cv2
import numpy as np
file_name = "/path/to/input/img/Unbenannt.jpg" # can be also .png
img = cv2.imread(file_name)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
th, threshed = cv2.threshold(gray, 240, 255, cv2.THRESH_BINARY_INV)
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (11,11))
morphed = cv2.morphologyEx(threshed, cv2.MORPH_CLOSE, kernel)
_, roi, _ = cv2.findContours(morphed, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
mask = np.zeros(img.shape, img.dtype)
cv2.fillPoly(mask, roi, (255,)*img.shape[2], )
masked_image = cv2.bitwise_and(img, mask)
cv2.imwrite("/path/to/output/mask_test.png", masked_image)
Вход:
Текущий выход:
Как уже упоминалось, я хочу сделать фон прозрачным.
Помощь очень ценится.
Привет, спасибо за ответ. да видел это раньше. Но это все равно не сработало. Во всяком случае, я разобрался в этом тем временем и опубликую решение здесь, после фазы экзамена (~ середина февраля);)





Я обнаружил, что мне нужно преобразовать изображение в BGRA, чтобы получить прозрачный фон. Я также добавил метод обрезки изображения до его ограничивающего прямоугольника. Как и обещал, рабочий код:
import cv2
import numpy as np
file_name = "/path/to/img.png"
def cut(img):
# crop image
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
th, threshed = cv2.threshold(gray, 240, 255, cv2.THRESH_BINARY_INV)
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (11,11))
morphed = cv2.morphologyEx(threshed, cv2.MORPH_CLOSE, kernel)
_, cnts, _ = cv2.findContours(morphed, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnt = sorted(cnts, key=cv2.contourArea)[-1]
x,y,w,h = cv2.boundingRect(cnt)
new_img = img[y:y+h, x:x+w]
return new_img
def transBg(img):
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
th, threshed = cv2.threshold(gray, 240, 255, cv2.THRESH_BINARY_INV)
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (11,11))
morphed = cv2.morphologyEx(threshed, cv2.MORPH_CLOSE, kernel)
_, roi, _ = cv2.findContours(morphed, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
mask = np.zeros(img.shape, img.dtype)
cv2.fillPoly(mask, roi, (255,)*img.shape[2], )
masked_image = cv2.bitwise_and(img, mask)
return masked_image
def fourChannels(img):
height, width, channels = img.shape
if channels < 4:
new_img = cv2.cvtColor(img, cv2.COLOR_BGR2BGRA)
return new_img
return img
s_img = cv2.imread(file_name, -1)
# set to 4 channels
s_img = fourChannels(s_img)
# remove white background
s_img = cut(s_img)
# set background transparent
s_img = transBg(s_img)
cv2.imwrite("/path/to/store/img.png", s_img)
ввод:
вывод - изображение с прозрачным фоном:
Для более новых версий OpenCV
import cv2
import numpy as np
file_name = "/path/to/img.png"
def cut(img):
# crop image
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
th, threshed = cv2.threshold(gray, 240, 255, cv2.THRESH_BINARY_INV)
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (11,11))
morphed = cv2.morphologyEx(threshed, cv2.MORPH_CLOSE, kernel)
cnts, _ = cv2.findContours(morphed, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnt = sorted(cnts, key=cv2.contourArea)[-1]
x,y,w,h = cv2.boundingRect(cnt)
new_img = img[y:y+h, x:x+w]
return new_img
def transBg(img):
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
th, threshed = cv2.threshold(gray, 240, 255, cv2.THRESH_BINARY_INV)
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (11,11))
morphed = cv2.morphologyEx(threshed, cv2.MORPH_CLOSE, kernel)
roi, _ = cv2.findContours(morphed, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
mask = np.zeros(img.shape, img.dtype)
cv2.fillPoly(mask, roi, (255,)*img.shape[2], )
masked_image = cv2.bitwise_and(img, mask)
return masked_image
def fourChannels(img):
height, width, channels = img.shape
if channels < 4:
new_img = cv2.cvtColor(img, cv2.COLOR_BGR2BGRA)
return new_img
return img
s_img = cv2.imread(file_name, -1)
# set to 4 channels
s_img = fourChannels(s_img)
# remove white background
s_img = cut(s_img)
# set background transparent
s_img = transBg(s_img)
cv2.imwrite("/path/to/store/img.png", s_img)
Вы пробовали этот stackoverflow.com/questions/36921496/…?