Маскировка определенного региона в OpenCV Python - PullRequest
0 голосов
/ 29 июня 2019

У меня есть изображение фермы коров. На изображении есть две области интереса (ROI). Из каждого ROI я хочу, чтобы все было черным. enter image description here

Координаты каждого угла области интереса -

1= [0, 1440]
2= [0, 1087]
3= [977, 80]
4= [1925, 67]
5= [2560, 800]
6= [2560, 1440]
7= [1465, 1440]
8= [1455,60]

Я использую следующие коды, чтобы замаскировать красную область и сделать все черным из области интереса.

import cv2, numpy as np
original_frame = cv2.imread("original.jpg")
frame = original_frame.copy()

# pts - location of the corners of the roi
pts = np.array([[0, 1450], [0, 1087], [977, 80], [1925, 67], [2560, 800], [2560, 1440]])
(x,y,w,h) = cv2.boundingRect(pts)

pts = pts - pts.min(axis=0)
mask = np.zeros(original_frame.shape, np.uint8)
cv2.drawContours(mask, [pts], -1, (255, 255, 255), -1, cv2.LINE_AA)
result = cv2.bitwise_and(original_frame, mask)

cv2.imwrite("out.jpg", result)

Результат довольно хороший, но все же покрывает некоторую дополнительную область сверху.

enter image description here

Если я попытаюсь замаскировать синюю область, изменив

pts = np.array([[1455,60], [1925, 67], [2560, 800], [2560, 1440],[1465, 1440] ])

Я получаю совершенно неверный результат enter image description here

Есть ли способ получить правильный результат для синего ROI?

1 Ответ

0 голосов
/ 29 июня 2019

Я делал весь процесс сложным.Существует очень прямой и простой ответ.

import numpy as np
import cv2
import matplotlib.pyplot as plt

# create a polygons using all outer corners of the ROI
external_poly = np.array( [[[1458,1440],[0,1440],[0,0],[2560,0],[2560,740], [1940,60], [1453,60]]], dtype=np.int32 )
im = cv2.imread("original.png", 1)
cv2.fillPoly( im , external_poly, (0,0,0) )
cv2.imwrite("output.jpg", im) 

Этот метод создает многоугольник, который берет все за пределами области интереса и заполняет многоугольник черным цветом.enter image description here

...