本文實例為大家分享了python實現摳圖的具體代碼,供大家參考,具體內容如下
其中使用了opencv中的grabcut方法
直接上代碼
# encoding:utf-8# 圖像提取# create by import numpy as npimport cv2from matplotlib import pyplot as plt img = cv2.imread('1.jpg')mask = np.zeros(img.shape[:2], np.uint8) bgdModel = np.zeros((1, 65), np.float64)fgdModel = np.zeros((1, 65), np.float64) rect = (20, 20, 413, 591)cv2.grabCut(img, mask, rect, bgdModel, fgdModel, 10, cv2.GC_INIT_WITH_RECT) mask2 = np.where((mask == 2) | (mask == 0), 0, 1).astype('uint8')img = img * mask2[:, :, np.newaxis]img += 255 * (1 - cv2.cvtColor(mask2, cv2.COLOR_GRAY2BGR))# plt.imshow(img)# plt.show()img = np.array(img)mean = np.mean(img)img = img - meanimg = img * 0.9 + mean * 0.9img /= 255plt.imshow(img)plt.show()
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持武林站長站。
新聞熱點
疑難解答