Ejemplo n.º 1
0
    def find(self, path, numThreshold, numInliers, numParabolaPoints,
             numIteractions):

        imageHandler = Image()
        hough = Hough()

        image, edges, nonzeros = imageHandler.getEdges(path)

        angle = hough.findAndClearLines(image, edges, 250)

        edges = imageHandler.rotate(angle, edges, nonzeros)

        image, equation = findParabola(image, edges, numThreshold, numInliers,
                                       numParabolaPoints, numIteractions)

        plotAndRotateParabola(image, equation, angle * -1)

        cv2.imwrite(path, image)