Как я могу нарисовать прямоугольник из точек, которые я хочу, используя ROI? - PullRequest
Новая
       14

Как я могу нарисовать прямоугольник из точек, которые я хочу, используя ROI?

0 голосов
/ 14 сентября 2018

Здравствуйте, я новичок в OpenCv. У меня есть изображение лабиринта. Я написал код решателя лабиринта. Мне нужно, чтобы фото, как на картинке, работало. Я хочу выбрать контуры белой области, используя ROI, но я не смог

Когда я пробую метод ROI, я получаю гладкий прямоугольник с выделенной черной областью.

https://i.stack.imgur.com/Ty5BX.png -----> это мой код результата

https://i.stack.imgur.com/S7zuJ.png --------> Я хочу этот результат 

import cv2
import numpy as np

#import image
image = cv2.imread('rt4.png')

#grayscaleqq
gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
#cv2.imshow('gray', gray)
#qcv2.waitKey(0)

#binary
#ret,thresh = cv2.threshold(gray,127,255,cv2.THRESH_BINARY_INV)
threshold = 150
thresh = cv2.threshold(gray, threshold, 255, cv2.THRESH_BINARY)[1]
cv2.namedWindow('second', cv2.WINDOW_NORMAL)
cv2.imshow('second', thresh)
cv2.waitKey(0)
cv2.destroyAllWindows()

#dilation
kernel = np.ones((1,1), np.uint8)
img_dilation = cv2.dilate(thresh, kernel, iterations=1)
cv2.namedWindow('dilated', cv2.WINDOW_NORMAL)
cv2.imshow('dilated', img_dilation)
cv2.waitKey(0)
cv2.destroyAllWindows()




#find contours
im2,ctrs, hier = cv2.findContours(img_dilation.copy(), 
cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)


#sort contours
sorted_ctrs = sorted(ctrs, key=lambda ctr: cv2.boundingRect(ctr) 
[0])

list = []

for i, ctr in enumerate(sorted_ctrs):
# Get bounding box
x, y, w, h = cv2.boundingRect(ctr)

# Getting ROI
roi = image[y:y+h, x:x+w]

a = w-x
b = h-y
list.append((a,b,x,y,w,h))


# show ROI
#cv2.imshow('segment no:'+str(i),roi)
cv2.rectangle(image,(x,y),( x + w, y + h ),(0,255,0),2)
#cv2.waitKey(0)

if w > 15 and h > 15:
    cv2.imwrite('home/Desktop/output/{}.png'.format(i), roi)

cv2.namedWindow('marked areas', cv2.WINDOW_NORMAL)
cv2.imshow('marked areas',image)
cv2.waitKey(0)
cv2.destroyAllWindows()




gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)

gray = np.float32(gray)
dst = cv2.cornerHarris(gray,2,3,0.04)

#result is dilated for marking the corners, not important
dst = cv2.dilate(dst,None)


image[dst>0.01*dst.max()]=[0,0,255]

cv2.imshow('dst',image)
if cv2.waitKey(0) & 0xff == 27:
cv2.destroyAllWindows()

list.sort()
print(list[len(list)-1])

Ответы [ 2 ]

0 голосов
/ 14 сентября 2018

Я неправильно понял ваш вопрос ранее. Итак, я переписываю.

Как уже сказал @Silencer, вы можете использовать метод drawContours. Вы можете сделать это следующим образом: 

import cv2
import numpy as np

#import image
im = cv2.imread('Maze2.png')
gaus = cv2.GaussianBlur(im, (5, 5), 1)
# mask1 = cv2.dilate(gaus, np.ones((15, 15), np.uint8, 3))
mask2 = cv2.erode(gaus, np.ones((5, 5), np.uint8, 1))
imgray = cv2.cvtColor(mask2, cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(imgray, 127, 255, 0)
im2, contours, hierarchy = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

maxArea1=0
maxI1=0

for i in range(len(contours)):
    area = cv2.contourArea(contours[i])
    epsilon = 0.01 * cv2.arcLength(contours[i], True)
    approx = cv2.approxPolyDP(contours[i], epsilon, True)
    if area > maxArea1 :
        maxArea1 = area

print(maxArea1)
print(maxI1)
cv2.drawContours(im, contours, maxI1, (0,255,255), 3)

cv2.imshow("yay",im)
cv2.imshow("gray",imgray)
cv2.waitKey(0)

cv2.destroyAllWindows()

Я использовал его на следующем изображении: enter image description here И я получил правильный ответ. Вы можете добавить дополнительные фильтры или уменьшить область, используя ROI, чтобы уменьшить расхождение, но это не требуется

Надеюсь, это поможет!

0 голосов
/ 14 сентября 2018

Простым решением просто нарисовать наклонный прямоугольник было бы использовать cv2.polylines. Исходя из вашего результата, я предполагаю, что у вас уже есть координаты вершин области, давайте назовем их [x1, y1], [x2, y2], [x3, y3], [x4, y4]. Функция polylines рисует линию от вершины к вершине, чтобы создать замкнутый многоугольник. 

import cv2
import numpy as np

#List coordinates of vertices as an array
pts = np.array([[x1,y1],[x2,y2],[x3,y3],[x4,y4]], np.int32)
pts = pts.reshape((-1,1,2))

#Draw lines from vertex to vertex
cv2.polylines(image, [pts], True, (255,0,0))
...