Определить расстояние между линиями

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

(эта строка line выглядит так: this )

Затем я отображаю только параллельные, но я не знаю, как определить минимальное и максимальное значение расстояния (в пикселях).

Любая помощь, пожалуйста?

Вот что я написал до сих пор:

import numpy as np
import cv2
import time
import math
start_time = time.time()

base = cv2.imread('prova/eq.jpg')
gray = cv2.cvtColor(base, cv2.COLOR_BGR2GRAY)
img = cv2.bilateralFilter(gray, 9, 90, 1)

kernel = np.ones((2,2), np.uint8)
img_dilation = cv2.dilate(img, kernel, iterations=2)
img_erosion = cv2.erode(img, kernel, iterations=4)

edges = cv2.Canny(img_erosion,cv2.THRESH_BINARY+cv2.THRESH_OTSU,550,apertureSize = 3)
#edges = cv2.Canny(gray,1,250,apertureSize = 3)


cv2.imwrite('edges.jpg',edges)


minLineLength=50
maxLineGap=10
threshold=20

image=edges

lines = cv2.HoughLinesP(image=image,rho=1,theta=np.pi/180, threshold=threshold,lines=np.array([]), minLineLength=minLineLength,maxLineGap=maxLineGap)

angles = []
for x1,y1,x2,y2 in lines[0]:
    theta = math.atan(float(y2 - y1) / (x2 - x1))
    angles.append({'theta': theta, 'p1': (x1, y1), 'p2': (x2, y2)})

for i in range(len(angles)):
    for j in range(i + 1, len(angles)):
        delta = abs(angles[i]['theta'] - angles[j]['theta'])
        if delta < 0.01:
            print(delta)#, angles[i], angles[j])
            cv2.line(base, angles[i]['p1'], angles[i]['p2'], (0, 0, 255), 2, cv2.CV_AA)
            cv2.line(base, angles[j]['p1'], angles[j]['p2'], (255, 0, 0), 2, cv2.CV_AA)






a,b,c = lines.shape
#for i in range(a):
#    cv2.line(base, (lines[i][0][0], lines[i][0][1]), (lines[i][0][2], lines[i][0][3]), (0, 0, 255), 13, cv2.CV_AA)

cv2.imwrite('prova/prova.jpg',base)


print("--- %s seconds ---" % (time.time() - start_time))

python opencv math line canny-operator
person Community    schedule 28.04.2017    source источник

Ответы (2)


arrow_upward
1
arrow_downward

Это то, что я написал, теперь должно работать

import numpy as np
import cv2
import time
import math
from PIL import Image

start_time = time.time()
base = cv2.imread('img/2009/no.jpg')
#goldouzian5 //  karimzadech4 // kitatani5 watanabe4   baccala5
gray = cv2.cvtColor(base, cv2.COLOR_BGR2GRAY)
img = cv2.bilateralFilter(gray, 11, 80, 11)

kernel = np.ones((2,2), np.uint8)
img_dilation = cv2.dilate(img, kernel, iterations=2)
img_erosion = cv2.erode(img, kernel, iterations=5)



edges = cv2.Canny(img_erosion,cv2.THRESH_BINARY+cv2.THRESH_OTSU,550,apertureSize = 3)
cv2.imwrite('edges.jpg',edges)

base = base



minLineLength=50
maxLineGap=11
threshold=25
image=edges

lines = cv2.HoughLinesP(image=image,rho=1,theta=np.pi/180, threshold=threshold,lines=np.array([]), minLineLength=minLineLength,maxLineGap=maxLineGap)
print("--- %s seconds (hough) ---" % (time.time() - start_time))

def segmentProjectedLength(p1, p2, q1, q2, theta):
    rot = lambda p: (p[0] * math.cos(theta) + p[1] * math.sin(theta), p[0] * math.sin(theta) - p[1] * math.cos(theta))
    p1 = rot(p1)
    p2 = rot(p2)
    q1 = rot(q1)
    q2 = rot(q2)

    minP = min(p1[0], p2[0])
    maxP = max(p1[0], p2[0])
    minQ = min(q1[0], q2[0])
    maxQ = max(q1[0], q2[0])

    return (min(maxP, maxQ) - max(minP, minQ)) if maxP >= minQ and maxQ >= minP else 0

lines = lines[0] if lines is not None else []
angles = []
for x1,y1,x2,y2 in lines:
    theta = math.atan(float(y2 - y1) / (x2 - x1)) if (x2 != x1) else math.pi / 2
    Rho = (x1 * y2 - x2 * y1) / math.sqrt((x2 - x1) ** 2 + (y2 - y1) ** 2)
    angles.append({'theta': theta, 'p1': (x1, y1), 'p2': (x2, y2), 'rho': Rho})
    #cv2.line(base, (x1, y1), (x2, y2), (0, 255, 0), 2, cv2.CV_AA)
print('--- %d lines found ---' % len(angles))
for i in range(len(angles)):
    for j in range(i + 1, len(angles)):

        delta = abs(angles[i]['theta'] - angles[j]['theta'])
        dist = abs(angles[i]['rho'] - angles[j]['rho'])
        projLength = segmentProjectedLength(angles[i]['p1'], angles[i]['p2'], angles[j]['p1'], angles[j]['p2'], angles[j]['theta'])

        if delta < 0.05 and dist > 5 and dist < 20 and projLength > 5:

                #print  (dist)
                cv2.line(base, angles[i]['p1'], angles[i]['p2'], (0, 0, 255), 2, cv2.CV_AA)
                cv2.line(base, angles[j]['p1'], angles[j]['p2'], (255, 0, 0), 2, cv2.CV_AA)
cv2.imwrite('prova/prova.jpg',base)
print("--- %s seconds (complete) ---" % (time.time() - start_time))
person Community    schedule 02.05.2017

arrow_upward
0
arrow_downward

Для параллельных прямых (одинаковый угол) абсолютная rho разница равна расстоянию между линиями.

Dist = |line1.rho - line2.rho|

Если нет доступа к параметрам 'rho' для строк (например, функция HoughLines без P возвращает), вычислить его с помощью

Len = Sqrt((x2-x1)^2 + (y2-y1)^2)
Rho = (x1 * y2 - x2 * y1) / Len
person MBo    schedule 28.04.2017
comment
Не могли бы вы помочь мне написать это на Python? - person ; 28.04.2017
comment
Я недостаточно знаю питон. ^2 в моей записи в квадрате - person MBo; 28.04.2017