HYTOS

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

    app = QApplication(sys.argv)

    try:

        if 1 == len(sys.argv):

            options = QFileDialog.Options()

            options |= QFileDialog.DontUseNativeDialog

            path, _ = QFileDialog.getOpenFileName(None, "Select Image Drawing File", os.getcwd(), "Image Files (*.png *.jpg)", options=options)

        elif 2 == len(sys.argv):

            path = sys.argv[1]  # 1 argument given is a string for the path of drawing

        #region Program start

        main(path)

        #endregion

    except Exception as ex:

        print('에러가 발생했습니다.\n', ex)

    sys.exit(app.exec_())

import cv2

import numpy as np

import matplotlib.pyplot as plt

import math

import imutils

from shapely.geometry import LineString

from shapely.ops import linemerge

class LineVector:

    def __init__(self, x1, y1, x2, y2):

        self.x1 = x1

        self.y1 = y1

        self.x2 = x2

        self.y2 = y2

        dx = x2 - x1

        dy = y2 - y1

        self.Length = math.sqrt(dx*dx + dy*dy)

        self.dx = dx/self.Length

        self.dy = dy/self.Length

        self.DrawFlag = True

    """

    두 직선이 평행한지 판별한다.

    """

    def IsParallel(self, rhs):

        dx = math.fabs(self.dx) - math.fabs(rhs.dx)

        dy = math.fabs(self.dy) - math.fabs(rhs.dy)

        Length = math.sqrt(dx*dx + dy*dy)

        cross = []

        dx1 = rhs.x1 - self.x1

        dy1 = rhs.y1 - self.y1

        dx2 = rhs.x1 - self.x2

        dy2 = rhs.y1 - self.y2

        cross.append(dx1*dy1 - dy1*dx2)

        dx1 = rhs.x2 - self.x1

        dy1 = rhs.y2 - self.y1

        dx2 = rhs.x2 - self.x2

        dy2 = rhs.y2 - self.y2

        cross.append(dx1*dy1 - dy1*dx2)

        if cross[0] > 0 and cross[1] > 0: return False

        return Length < 0.0001

    """

    두 직선의 거리를 구한다.

    """

    def DistanceTo(self, rhs):

        dx = rhs.x1 - self.x1

        dy = rhs.y1 - self.y1

        dot = self.dx*dx + self.dy*dy

        tmp = dot

        x1 = self.x1 + tmp*self.dx

        y1 = self.y1 + tmp*self.dy

        dx = x1 - rhs.x1

        dy = y1 - rhs.y1

        return math.sqrt(dx*dx + dy*dy)

    '''

    두 직선의 사이를 가로지르는 직선을 구한다.

    '''

    def Merge(self, rhs):

        dx1 = rhs.x1 - self.x1

        dy1 = rhs.y1 - self.y1

        dx2 = rhs.x1 - self.x2

        dy2 = rhs.y1 - self.y2

        candidates = []

        candidates.append((self.x1, self.y1))

        candidates.append((self.x2, self.y2))

        Dir = []

        cross = dx1*dy2 - dy1*dx2

        if cross > 0:

            dot = dx1*self.dx + dy1*self.dy

            tmp = (self.x1 + dot*self.dx, self.y1 + dot*self.dy)

            candidates.append(tmp)

            Dir.append((rhs.x1 - tmp[0], rhs.y1 - tmp[1]))

        dx1 = rhs.x2 - self.x1

        dy1 = rhs.y2 - self.y1

        dx2 = rhs.x2 - self.x2

        dy2 = rhs.y2 - self.y2

        cross = dx1*dy2 - dy1*dx2

        if cross > 0:

            dot = dx1*self.dx + dy1*self.dy

            tmp = (self.x1 + dot*self.dx, self.y1 + dot*self.dy)

            candidates.append(tmp)

            Dir.append((rhs.x2 - tmp[0], rhs.y2 - tmp[1]))

        MaxLength = None

        Points = []

        for i in range(len(candidates) - 1):

            for j in range(i+1, len(candidates)):

                dx = candidates[i][0] - candidates[j][0]

                dy = candidates[i][1] - candidates[j][1]

                Length = dx*dx + dy*dy

                if MaxLength == None:

                    MaxLength = Length

                    Points.append(candidates[i])

                    Points.append(candidates[j])

                elif MaxLength < Length:

                    MaxLength = Length

                    Points[0] = candidates[i]

                    Points[1] = candidates[j]

        if len(Points) == 2 and len(Dir) > 0:

            x1 = Points[0][0] + Dir[0][0]*0.5

            y1 = Points[0][1] + Dir[0][1]*0.5

            x2 = Points[1][0] + Dir[0][0]*0.5

            y2 = Points[1][1] + Dir[0][1]*0.5

            return LineVector(x1, y1, x2, y2)

        else:

            dx1 = rhs.x1 - self.x1

            dy1 = rhs.y1 - self.y1

            dot = dx1*self.dx + dy1*self.dy

            tmp = (self.x1 + dot*self.dx, self.y1 + dot*self.dy)

            Dir.append((rhs.x1 - tmp[0], rhs.y1 - tmp[1]))

            x1 = Points[0][0] + Dir[0][0]*0.5

            y1 = Points[0][1] + Dir[0][1]*0.5

            x2 = Points[1][0] + Dir[0][0]*0.5

            y2 = Points[1][1] + Dir[0][1]*0.5

            return LineVector(x1, y1, x2, y2)

        return LineVector(self.x1, self.y1, self.x2, self.y2)

    def Draw(self, img, r, g, b):

        cv2.line(img, (int(self.x1), int(self.y1)), (int(self.x2), int(self.y2)), (r, g, b), 1)

class LineDetector():

    def __init__(self, image):

        thresh = 127

        self.Image = cv2.threshold(image, thresh, 255, cv2.THRESH_BINARY)[1]

        w, h = image.shape[::-1]

        self.Result = np.zeros((h,w,3), np.uint8)

    ##

    ## 이미지에서 직선을 검출한다.

    ##

    def Detect(self):

        kernel = np.ones((2, 2), np.uint8)

        #erosion = cv2.dilate(self.Image, kernel, iterations=2)

        erosion = cv2.morphologyEx(self.Image, cv2.MORPH_ERODE, kernel)

        #erosion = self.Image

        edge = cv2.bitwise_not(erosion)

        #edge = cv2.Canny(erosion, 250, 500)

        lines = cv2.HoughLinesP(edge, 1, np.pi/180, 500, 500, 100)

        LineVectors = []

        '''

        LineVectors.append(LineVector(1984, 612, 1984, 987))

        LineVectors.append(LineVector(1989, 615, 1989, 870))

        '''

        for line in lines:

            x1, y1, x2, y2 = line[0]

            line = LineString([(x1, y1), (x2, y2)])

            #line = LineVector(x1, y1, x2, y2)

            LineVectors.append(line)

        '''

        MergedLines = []

        for i in range(len(LineVectors)):

            if LineVectors[i].DrawFlag == False: continue

            for j in range(i+1, len(LineVectors)):

                if LineVectors[j].DrawFlag == False: continue

                if LineVectors[i].IsParallel(LineVectors[j]) and (LineVectors[i].DistanceTo(LineVectors[j]) < 10):

                    MergedLines.append(LineVectors[i].Merge(LineVectors[j]))

                    Merged = True

                    LineVectors[i].DrawFlag = False

                    LineVectors[j].DrawFlag = False

                    break

            if LineVectors[i].DrawFlag == True: MergedLines.append(LineVectors[i])

        '''

        #lines = linemerge(LineVectors)

        #for line in list(lines):

        #    line.Draw(self.Result, 0, 255, 0)

        '''

        for line in MergedLines:

            line.Draw(self.Result, 0, 0, 255)

        '''

        #contours는 point의 list형태. 예제에서는 사각형이 하나의 contours line을 구성하기 때문에 len(contours) = 1. 값은 사각형의 꼭지점 좌표.

        #hierachy는 contours line의 계층 구조

        image, contours, hierachy = cv2.findContours(erosion, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

        image = cv2.drawContours(self.Result, contours, -1, (0,255,0), 3)

        #b, g, r = cv2.split(src)

        #pltSrc = cv2.merge([r, g, b])

        #plt.imshow(pltSrc)

        #plt.show()

        #plt.imshow(erosion)

        #plt.show()

        cv2.imwrite("d:/Projects/DTIPID/DTI_PID/DTI_PID/res/lines.png", self.Result)

        #resized = cv2.resize(edge, None, fx = 0.25, fy = 0.25)

        #cv2.imshow('Canny', resized)

        #cv2.imshow('Canny', erosion)

from LineDetector import LineDetector

    #lines = linedraw.sketch(path)

    #linedraw.visualize(lines)

    #imgLines = passpotrace(path)    #execpath + '\\res\\UY1-K-2007_P1_300dpi_black.png')

    image  = cv2.imread(path, cv2.IMREAD_GRAYSCALE)

    lineDetector = LineDetector(image)

    lineDetector.Detect()

    try:

        xmlFilePath = os.path.dirname(path) + '\\' + os.path.basename(path).split('.')[0] + '.xml'

        file = open(xmlFilePath, 'w')

        for line in imgLines:

            file.write(str(line) + '\n')

        file.close()

    except Exception as ex:

        print('에러가 발생했습니다.\n', ex)