我想弄清楚如何洗掉我已經生成的選定區域的背景影像。為了清楚地理解,我附上了一個關于我需要解決的問題的可視化。
黑色區域是被選中的區域,我想把它反轉。所以選中的區域會變成彩色圖片,背景會是黑色。
這是我已經創建的簡單代碼:
import cv2
import numpy as np
pic= cv2.imread('dataset/set.jpeg')
pic = cv2.resize(pic, dsize=(500, 400), interpolation=cv2.INTER_CUBIC)
gray=cv2.cvtColor(pic,cv2.COLOR_BGR2GRAY)
blur = cv2.GaussianBlur(gray,(5,5),5)
_,thres = cv2.threshold(blur, 100,250, cv2.THRESH_TOZERO)
res = cv2.Canny(thres, 100, 200, L2gradient=True)
circles = cv2.HoughCircles(res,cv2.HOUGH_GRADIENT,1,20,param1=200,param2=15,minRadius=80,maxRadius=100)
for i in circles[0,:]:
# draw the outer circle
cv2.circle(pic,(int(i[0]),int(i[1])),int(i[2]),(255,255,255),2)
i = i.astype(int)
pic[i[1]-i[2]:i[1] i[2], i[0]-i[2]:i[0] i[2]] = 0
cv2.imshow('Hole',pic)
cv2.waitKey(0)
cv2.destroyAllWindows()
問題出在這段代碼中:
pic[i[1]-i[2]:i[1] i[2], i[0]-i[2]:i[0] i[2]] = 0
我正在將選定區域更改為黑色(零),但我不知道如何將其反轉為唯一有顏色的區域,而讓另一個區域成為黑色影像。請幫我解決這個問題,謝謝!!
uj5u.com熱心網友回復:
該問題已通過創建蒙版并通過以下代碼行組合前景和背景來解決:
import cv2
import numpy as np
pic= cv2.imread('Assets/set.jpeg')
pic = cv2.resize(pic, dsize=(500, 400), interpolation=cv2.INTER_CUBIC)
gray=cv2.cvtColor(pic,cv2.COLOR_BGR2GRAY)
blur = cv2.GaussianBlur(gray,(5,5),5)
_,thres = cv2.threshold(blur, 100,250, cv2.THRESH_TOZERO)
res = cv2.Canny(thres, 100, 250, L2gradient=True)
circles = cv2.HoughCircles(res,cv2.HOUGH_GRADIENT,1,20,param1=200,param2=15,minRadius=80,maxRadius=100)
circles = np.uint16(np.around(circles))
mask = np.full((res.shape[0], res.shape[1]), 1, dtype=np.uint8) # mask is only
clone = pic.copy()
for i in circles[0, :]:
cv2.circle(mask, (i[0], i[1]), i[2], (255, 255, 255), -1)
cv2.circle(clone, (i[0], i[1]), i[2], (255, 255, 255), 1)
# get first masked value (foreground)
fg = cv2.bitwise_or(res, res, mask=mask)
# get second masked value (background) mask must be inverted
mask = cv2.bitwise_not(mask)
background = np.full(res.shape, 255, dtype=np.uint8)
bk = cv2.bitwise_or(background, background, mask=mask)
# combine foreground background
final = cv2.bitwise_or(fg, bk)
result = np.concatenate((res,final),axis=1)
cv2.imshow('Hole',result)
cv2.waitKey(0)
cv2.destroyAllWindows()
沒什么可問的了,我會結束這個問題。謝謝!!
轉載請註明出處,本文鏈接:https://www.uj5u.com/qiye/327247.html
上一篇:無法顯示影像opencv(c )
