2022年 11月 9日

python鼠标点击事件

 一、画矩形框

  1. import cv2
  2.  
  3. global img
  4. global point1, point2
  5. def on_mouse(event, x, y, flags, param):
  6.     global img, point1, point2
  7.     img2 = img.copy()
  8.     if event == cv2.EVENT_LBUTTONDOWN:         #左键点击
  9.         point1 = (x,y)
  10.         cv2.circle(img2, point1, 10, (0,255,0), 5)
  11.         cv2.imshow('image', img2)
  12.     elif event == cv2.EVENT_MOUSEMOVE and (flags & cv2.EVENT_FLAG_LBUTTON):               #按住左键拖曳
  13.         cv2.rectangle(img2, point1, (x,y), (255,0,0), 5)
  14.         cv2.imshow('image', img2)
  15.     elif event == cv2.EVENT_LBUTTONUP:         #左键释放
  16.         point2 = (x,y)
  17.         cv2.rectangle(img2, point1, point2, (0,0,255), 5) 
  18.         cv2.imshow('image', img2)
  19.         min_x = min(point1[0],point2[0])     
  20.         min_y = min(point1[1],point2[1])
  21.         width = abs(point1[0] - point2[0])
  22.         height = abs(point1[1] -point2[1])
  23.         cut_img = img[min_y:min_y+height, min_x:min_x+width]
  24.         cv2.imwrite('lena3.jpg', cut_img)
  25.  
  26. def main():
  27.     global img
  28.     img = cv2.imread('lena.jpg')
  29.     cv2.namedWindow('image')
  30.     cv2.setMouseCallback('image', on_mouse)
  31.     cv2.imshow('image', img)
  32.     cv2.waitKey(0)
  33.  
  34. if __name__ == '__main__':
  35.     main()

二、画线

  1. import os
  2.  
  3. import cv2
  4. import math
  5.  
  6. global img
  7. global point1, point2
  8.  
  9.  
  10. def calculate_distance(pt1, pt2):
  11.     x1, y1 = pt1
  12.     x2, y2 = pt2
  13.     dist = math.sqrt((x2 - x1) ** 2 + (y2 - y1) ** 2)
  14.     return dist
  15.  
  16.  
  17. def on_mouse(event, x, y, flags, param):
  18.     global img, point1, point2
  19.     img2 = img.copy()
  20.     if event == cv2.EVENT_LBUTTONDOWN:  # 左键点击
  21.         point1 = (x, y)
  22.         cv2.circle(img2, point1, 10, (0, 255, 0), 2)
  23.         cv2.imshow('image', img2)
  24.     elif event == cv2.EVENT_MOUSEMOVE and (flags & cv2.EVENT_FLAG_LBUTTON):  # 按住左键拖曳
  25.         cv2.line(img2, point1, (x, y), (255, 0, 0), 2)
  26.         cv2.imshow('image', img2)
  27.     elif event == cv2.EVENT_LBUTTONUP:  # 左键释放
  28.         point2 = (x, y)
  29.         cv2.line(img2, point1, point2, (0, 0, 255), 2)
  30.         cv2.imshow('image', img2)
  31.         distance = calculate_distance(point1, point2)
  32.         font = cv2.FONT_HERSHEY_SIMPLEX
  33.         cv2.putText(img2, f'{int(distance)}', (point2[0]+2, point2[1]+2), font, 1.2, (255, 255, 255), 2)
  34.         cv2.imshow('image', img2)
  35.         img = img2
  36.  
  37.  
  38. def main():
  39.     global img, base_name
  40.     img_path = r"D:\work\data\ellipse_fitting\ori_resize\label_viz0.png"
  41.     img_name = os.path.split(img_path)[-1]
  42.     base_name = os.path.splitext(img_name)[0]
  43.     save_name = base_name + '_gt.jpg'
  44.     save_dir = r'D:\work\data\ellipse_fitting\gt_axis'
  45.     os.makedirs(save_dir, exist_ok=True)
  46.  
  47.     img = cv2.imread(img_path)
  48.     cv2.namedWindow('image')
  49.     cv2.setMouseCallback('image', on_mouse)
  50.     cv2.imshow('image', img)
  51.     if cv2.waitKey(0) & 0xFF == ord('s'):
  52.         cv2.imwrite(os.path.join(save_dir, save_name), img)
  53.         print('save img:{}'.format(os.path.join(save_dir, save_name)))
  54.  
  55.  
  56. if __name__ == '__main__':
  57.     main()