0
0 комментариев

Задача:
Найти жёсткий диск на фото, определить его угол и контуры

Проблема:
Не всегда удаётся найти правильный контур диска.

В коде я делаю изображения серыми, блюрю, нахожу разницу между ними и определяю контуры, но решение не точное. Хотелось бы вашей помощи.

firstFrame = cv2.imread(bg_im)  # Фоновое изображение без диска
frame_img = cv2.imread(frame)   # Изображение с диском
 
gray = cv2.cvtColor(frame_img, cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray, (21, 21), 0)
firstFrame = cv2.cvtColor(firstFrame, cv2.COLOR_BGR2GRAY)
firstFrame = cv2.GaussianBlur(firstFrame, (21, 21), 0)
 
frameDelta = cv2.absdiff(firstFrame, gray)
thresh = cv2.threshold(frameDelta, 25, 255, cv2.THRESH_BINARY)[1]
thresh = cv2.dilate(thresh, None, iterations=2)
(cnts, _) = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
 
c = sorted(cnts, key = cv2.contourArea, reverse = True)[0]
 
rect = cv2.minAreaRect(c)
box = cv2.cv.BoxPoints(rect)
 
x1 = box[0][0]
x2 = box[1][0]
x4 = box[3][0]
 
y1 = box[0][1]
y2 = box[1][1]
y4 = box[3][1]
 
perimeter_1_4 = sqrt((x4 - x1)**2 + (y1 - y4)**2)
perimeter_1_2 = sqrt((x2 - x1)**2 + (y1 - y2)**2)
 
box = np.int0(box)
corner = list(rect)[2]
 
if perimeter_1_4 > perimeter_1_2:
    corner = -(90 - corner)
 
cv2.drawContours(frame_img,[box],0,(0,0,255),2)

Фон. Без диска
Фон. Без диска


Добавить комментарий