python鼠标点击事件

 一、画矩形框

import cv2
 
global img
global point1, point2
def on_mouse(event, x, y, flags, param):
    global img, point1, point2
    img2 = img.copy()
    if event == cv2.EVENT_LBUTTONDOWN:         #左键点击
        point1 = (x,y)
        cv2.circle(img2, point1, 10, (0,255,0), 5)
        cv2.imshow('image', img2)
    elif event == cv2.EVENT_MOUSEMOVE and (flags & cv2.EVENT_FLAG_LBUTTON):               #按住左键拖曳
        cv2.rectangle(img2, point1, (x,y), (255,0,0), 5)
        cv2.imshow('image', img2)
    elif event == cv2.EVENT_LBUTTONUP:         #左键释放
        point2 = (x,y)
        cv2.rectangle(img2, point1, point2, (0,0,255), 5) 
        cv2.imshow('image', img2)
        min_x = min(point1[0],point2[0])     
        min_y = min(point1[1],point2[1])
        width = abs(point1[0] - point2[0])
        height = abs(point1[1] -point2[1])
        cut_img = img[min_y:min_y+height, min_x:min_x+width]
        cv2.imwrite('lena3.jpg', cut_img)
 
def main():
    global img
    img = cv2.imread('lena.jpg')
    cv2.namedWindow('image')
    cv2.setMouseCallback('image', on_mouse)
    cv2.imshow('image', img)
    cv2.waitKey(0)
 
if __name__ == '__main__':
    main()

二、画线

import os

import cv2
import math

global img
global point1, point2


def calculate_distance(pt1, pt2):
    x1, y1 = pt1
    x2, y2 = pt2
    dist = math.sqrt((x2 - x1) ** 2 + (y2 - y1) ** 2)
    return dist


def on_mouse(event, x, y, flags, param):
    global img, point1, point2
    img2 = img.copy()
    if event == cv2.EVENT_LBUTTONDOWN:  # 左键点击
        point1 = (x, y)
        cv2.circle(img2, point1, 10, (0, 255, 0), 2)
        cv2.imshow('image', img2)
    elif event == cv2.EVENT_MOUSEMOVE and (flags & cv2.EVENT_FLAG_LBUTTON):  # 按住左键拖曳
        cv2.line(img2, point1, (x, y), (255, 0, 0), 2)
        cv2.imshow('image', img2)
    elif event == cv2.EVENT_LBUTTONUP:  # 左键释放
        point2 = (x, y)
        cv2.line(img2, point1, point2, (0, 0, 255), 2)
        cv2.imshow('image', img2)
        distance = calculate_distance(point1, point2)
        font = cv2.FONT_HERSHEY_SIMPLEX
        cv2.putText(img2, f'{int(distance)}', (point2[0]+2, point2[1]+2), font, 1.2, (255, 255, 255), 2)
        cv2.imshow('image', img2)
        img = img2


def main():
    global img, base_name
    img_path = r"D:\work\data\ellipse_fitting\ori_resize\label_viz0.png"
    img_name = os.path.split(img_path)[-1]
    base_name = os.path.splitext(img_name)[0]
    save_name = base_name + '_gt.jpg'
    save_dir = r'D:\work\data\ellipse_fitting\gt_axis'
    os.makedirs(save_dir, exist_ok=True)

    img = cv2.imread(img_path)
    cv2.namedWindow('image')
    cv2.setMouseCallback('image', on_mouse)
    cv2.imshow('image', img)
    if cv2.waitKey(0) & 0xFF == ord('s'):
        cv2.imwrite(os.path.join(save_dir, save_name), img)
        print('save img:{}'.format(os.path.join(save_dir, save_name)))


if __name__ == '__main__':
    main()