/Python/Draw line from 2 mouse event in image python

Draw line from 2 mouse event in image python

import cv2

class DrawLineWidget(object):
    def __init__(self):
        self.original_image = cv2.imread('1.jpg')
        self.clone = self.original_image.copy()

        cv2.namedWindow('image')
        cv2.setMouseCallback('image', self.extract_coordinates)

        # List to store start/end points
        self.image_coordinates = []

    def extract_coordinates(self, event, x, y, flags, parameters):
        # Record starting (x,y) coordinates on left mouse button click
        if event == cv2.EVENT_LBUTTONDOWN:
            self.image_coordinates = [(x,y)]

        # Record ending (x,y) coordintes on left mouse bottom release
        elif event == cv2.EVENT_LBUTTONUP:
            self.image_coordinates.append((x,y))
            print('Starting: {}, Ending: {}'.format(self.image_coordinates[0], self.image_coordinates[1]))

            # Draw line
            cv2.line(self.clone, self.image_coordinates[0], self.image_coordinates[1], (36,255,12), 2)
            cv2.imshow("image", self.clone) 

        # Clear drawing boxes on right mouse button click
        elif event == cv2.EVENT_RBUTTONDOWN:
            self.clone = self.original_image.copy()

    def show_image(self):
        return self.clone

if __name__ == '__main__':
    draw_line_widget = DrawLineWidget()
    while True:
        cv2.imshow('image', draw_line_widget.show_image())
        key = cv2.waitKey(1)

        # Close program with keyboard 'q'
        if key == ord('q'):
            cv2.destroyAllWindows()
            exit(1)

Source: stackoverflow.com

Similar codes

1 vote399 views2 answers
Flask bootstrap
Python Flask
2 votes344 views3 answers
Can't find model 'en_core_web_sm'. It doesn't seem to be a shortcut link, a Python package or a valid path to a data directory.
Python
2 votes351 views2 answers
Cmd clear command for anaconda
Python
12 votes361 views6 answers
How to make a list python
Python
14 votes353 views1 answer
Pd.to_datetime python
Python
0 vote375 views1 answer
How to make a grading system in python
Python
0 vote354 views1 answer
Python Split list into chunks using for loop
Python
0 vote379 views1 answer
Creating base models django
Python
0 vote371 views1 answer
Name='resnet50_input', description="created by layer 'resnet50_input'") at layer "resnet50
Python
45 votes381 views6 answers
Get files in directory python
Python