python实现自动抠图
生活随笔
收集整理的這篇文章主要介紹了
python实现自动抠图
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
直接上圖
賬戶里有下載,有問題可私聊
import cv2 import numpy as npsrc = cv2.imread("9.jpg") src = cv2.resize(src, (0, 0), fx=0.5, fy=0.5)# 交互式,返回 (x_min, y_min, w, h) r = cv2.selectROI('input', src, True)# roi區域 roi = src[int(r[1]):int(r[1] + r[3]), int(r[0]):int(r[0] + r[2])]# 原圖mask,與原圖等大小 mask = np.zeros(src.shape[:2], dtype=np.uint8)# 矩形roi rect = (int(r[0]), int(r[1]), int(r[2]), int(r[3])) # 包括前景的矩形,格式為(x,y,w,h)# bg模型的臨時數組 bgdmodel = np.zeros((1, 65), np.float64) # fg模型的臨時數組 fgdmodel = np.zeros((1, 65), np.float64)cv2.grabCut(src, mask, rect, bgdmodel, fgdmodel, 11, mode=cv2.GC_INIT_WITH_RECT)print(np.unique(mask)) # 提取前景和可能的前景區域 mask2 = np.where((mask == 1) | (mask == 3), 255, 0).astype('uint8')print(mask2.shape)# 按位與 src & src == 0,得到的是二進制 result = cv2.bitwise_and(src, src, mask=mask2) # cv2.imwrite('result.jpg', result) # cv2.imwrite('roi.jpg', roi)cv2.imshow('mask', mask2) cv2.imshow('roi', roi) cv2.imshow("result", result) cv2.waitKey(0) cv2.destroyAllWindows()總結
以上是生活随笔為你收集整理的python实现自动抠图的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 用python实现背单词的小脚本系统
- 下一篇: snv报错