python - 如何在python中删除图像的背景
解决方案
这是使用 Python/OpenCV 的一种方法。
- 读取输入
- 转换为灰色
- 阈值和反转作为掩码
- 可选择应用形态学以清理任何多余的斑点
- 边缘抗锯齿
- 将输入的副本转换为 BGRA 并将遮罩插入为 Alpha 通道
- 保存结果
输入:
import cv2
import numpy as np
# load image
img = cv2.imread('person.png')
# convert to graky
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# threshold input image as mask
mask = cv2.threshold(gray, 250, 255, cv2.THRESH_BINARY)[1]
# negate mask
mask = 255 - mask
# apply morphology to remove isolated extraneous noise
# use borderconstant of black since foreground touches the edges
kernel = np.ones((3,3), np.uint8)
mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel)
mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel)
# anti-alias the mask -- blur then stretch
# blur alpha channel
mask = cv2.GaussianBlur(mask, (0,0), sigmaX=2, sigmaY=2, borderType = cv2.BORDER_DEFAULT)
# linear stretch so that 127.5 goes to 0, but 255 stays 255
mask = (2*(mask.astype(np.float32))-255.0).clip(0,255).astype(np.uint8)
# put mask into alpha channel
result = img.copy()
result = cv2.cvtColor(result, cv2.COLOR_BGR2BGRA)
result[:, :, 3] = mask
# save resulting masked image
cv2.imwrite('person_transp_bckgrnd.png', result)
# display result, though it won't show transparency
cv2.imshow("INPUT", img)
cv2.imshow("GRAY", gray)
cv2.imshow("MASK", mask)
cv2.imshow("RESULT", result)
cv2.waitKey(0)
cv2.destroyAllWindows()
透明结果:
推荐阅读
- elixir - Ecto - 多个条件不起作用。灵药
- ssrs-2016 - SQL Server 从标准升级到企业后,SSRS 报告卡在加载中
- regex - 在 10-K Edgar 填充物中使用 Beautiful Soup 和正则表达式提取文本
- highcharts - 如何在地图中移动岛屿
- html - 具有多种布局的 Flexbox 布局
- reactjs - 部署到 GitHub 页面时收到错误
- swift - 在隐式展开可选值时意外发现 nil
- perl - 无法执行 ESQL.EXE 来确定 ESQL/C 库
- arrays - 在 1 个 foreach 循环中运行两个数组
- c# - 使用 CreateFileAsync 创建的文件在哪里?