прозрачная маска на фото с координатами квадрата

Question

Мне нужно, из изображения с квадратом внутри и координатами, создать прозрачную маску. К сожалению, я ничего не нашел о литературе по прозрачности в openCV.

Например, у меня есть это фото:

С кодом ниже координаты для квадрата следующие

(237, 150, 311, 150, 74) - вот код

def loc(photo):
    img = cv2.imread(photo,cv2.IMREAD_COLOR)
    red = rgb2redChan(img)

    #read HAAR ade loop for find eye(s)
    try:
        eyes = eye_cascade.detectMultiScale(red)
        best_ex=eyes[0][0]
        best_ey=eyes[0][1]
        best_ew=eyes[0][2]
        best_eh=eyes[0][3]
        for (ex,ey,ew,eh) in eyes:
            if(best_eh<eh):
                best_ex=ex
                best_ey=ey
                best_ew=ew
                best_eh=eh
        print(ex,ey,ex+ew,ey,eh)
        cv2.rectangle(img,(best_ex,best_ey),(best_ex+best_ew,best_ey+best_eh),(255,255,255),2)

        img = createMask(img,best_ex,best_ey,best_ew,best_eh)
    except:
        print("Not localized")

    #show eye(s) rectangle in color image
    cv2.imshow('red',img)
    #press any key to close
    cv2.waitKey(0)
    cv2.destroyAllWindows()
    return img,photo.split("/")[-1]

def createMask(img,x,y,w,h):
    mask = np.zeros(img.shape,np.uint8)
    mask[y:y+h,x:x+w] = img[y:y+h,x:x+w]
    return mask

Я думаю, что проблема связана с созданием

mask = np.zeros(img.shape,np.uint8)

Все значения матрицы [0,0,0], как я могу превратить их в прозрачные значения?

P.S. начальное расширение фотографии: .jpg

Answer 1

Вы почти там, теперь используйте addWeighted

def createMask(img,x,y,w,h, alpha=0.5):
    '''
    update alpha to the darkness you want
    '''
    mask = np.zeros(img.shape,np.uint8)
    mask[y:y+h,x:x+w] = img[y:y+h,x:x+w]

    beta = 1 - alpha
    mask = cv2.addWeighted(img, alpha, mask, beta, 0)
    return mask

пример: