使用python来抠图

简介

本文主要介绍如何使用python来抠图,及抠图的基本原理

效果图

图片
原图 原图
效果图 原图

原理

  1. 首先要选定要抽取对象的区域,区域外的部分为背景
  2. 基于选定区域,grabCut循环处理前景内容
  3. 前景部分含有的背景(抽取对象外)部分,并不能很好处理。

代码

import cv2
import numpy as np

fname = 'images/test1.jpg'
img = cv2.imread(fname)
rect = (275, 120, 170, 320)

mask = np.zeros(img.shape[:2], np.uint8)
bgModel = np.zeros((1,65), np.float64)
fgModel = np.zeros((1,65), np.float64)
cv2.grabCut(img, mask, rect, bgModel, fgModel, 5, cv2.GC_INIT_WITH_RECT)
mask2 = np.where((mask == 2) | (mask == 0), 0, 1).astype(np.uint8)

out = img * mask2[:, :, np.newaxis]

cv2.imshow('output', out)
cv2.waitKey()

感言

  • 选定区域的width, height, x, y等参数的确定比较麻烦,且对结果影响较大
  • 当抽取不连续对象时,区域要包含所有对象
  • 当grabCut不能很好区分对象时,需要人工标记前景/背景部分。这时需要用到cv2.GC_INIT_WITH_MASK参数
  • 有兴趣的朋友可以设计一个在线抠图工具
    • 上传图片
    • 选定区域
    • 标记前景部分
    • 标记背景部分
    • 下载图片

参考链接

http://labs.eecs.tottori-u.ac.jp/sd/Member/oyamada/OpenCV/html/py_tutorials/py_imgproc/py_grabcut/py_grabcut.html

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注