Создание собственного контура в opencv с использованием python
У меня есть набор граничных точек объекта.
Я хочу нарисовать его, используя OpenCV в качестве контура.
Я понятия не имею, как конвертировать мои точки в контурное представление.
К тому же контурному представлению, которое получается следующим вызовом
contours,_ = cv2.findContours(image,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
Есть идеи?
Спасибо
3 ответа
Глядя на формат контуров, я думаю, что-то вроде этого должно быть достаточно:
contours = [numpy.array([[1,1],[10,50],[50,50]], dtype=numpy.int32) , numpy.array([[99,99],[99,60],[60,99]], dtype=numpy.int32)]
Эта небольшая программа дает работающий пример:
import numpy
import cv2
contours = [numpy.array([[1,1],[10,50],[50,50]], dtype=numpy.int32) , numpy.array([[99,99],[99,60],[60,99]], dtype=numpy.int32)]
drawing = numpy.zeros([100, 100],numpy.uint8)
for cnt in contours:
cv2.drawContours(drawing,[cnt],0,(255,255,255),2)
cv2.imshow('output',drawing)
cv2.waitKey(0)
Чтобы создать свой собственный контур из списка точек Python L
L=[[x1,y1],[x2,y2],[x3,y3],[x4,y4],[x5,y5],[x6,y6],[x7,y7],[x8,y8],[x9,y9],...[xn,yn]]
Создайте NTR-массив массива ctr из L, измените его и принудительно введите его тип
ctr = numpy.array(L).reshape((-1,1,2)).astype(numpy.int32)
ctr - наш новый граф, давайте нарисуем его на существующем изображении
cv2.drawContours(image,[ctr],0,(255,255,255),1)
Контур - это просто кривая, соединяющая все непрерывные точки, поэтому для создания собственного контура вы можете создать np.array()
с твоим (x,y)
точки по часовой стрелке
points = np.array([[25,25], [70,10], [150,50], [250,250], [100,350]])
Это оно!
В зависимости от того, что вам нужно, есть два метода рисования контура на изображении:
Контурное очертание
Если вам нужен только контур контура, используйте cv2.drawContours()
cv2.drawContours(image,[points],0,(0,0,0),2)
Заполненный контур
Чтобы получить заполненный контур, вы можете использовать cv2.fillPoly()
или cv2.drawContours()
с thickness=-1
cv2.fillPoly(image, [points], [0,0,0]) # OR
# cv2.drawContours(image,[points],0,(0,0,0),-1)
Полный пример кода для полноты
import cv2
import numpy as np
# Create blank white image
image = np.ones((400,400), dtype=np.uint8) * 255
# List of (x,y) points in clockwise order
points = np.array([[25,25], [70,10], [150,50], [250,250], [100,350]])
# Draw points onto image
cv2.drawContours(image,[points],0,(0,0,0),2)
# Fill points onto image
# cv2.fillPoly(image, [points], [0,0,0])
cv2.imshow('image', image)
cv2.waitKey()
Чтобы добавить к ответу Шерифа Каоуа, я обнаружил, что должен был преобразовать в список и сжать массив. Чтение в массиве точек из текстового файла:
contour = []
with open(array_of_points,'r') as f:
next(f) // line one of my file gives the number of points
for l in f:
row = l.split()
numbers = [int(n) for n in row]
contour.append(numbers)
ctr = np.array(contour).reshape((-1,1,2)).astype(np.int32)
ctr = ctr.tolist()
ctr = zip(*[iter(ctr)]*len(contour))