Python＆CV2：如何用鼠标在图像上绘制线条然后返回线条坐标？

Question

我正在画一条线，我正在逐帧地逐帧浏览，因此我可以计算线的角度。我做了一个非常简单的脚本，通过视频几步之遥，尝试收集到阵列中的每个图像中点击了点，但是连这似乎工作...下面的代码：

import cv2, cv 

cap = cv2.VideoCapture('video.avi') 

box = [] 
def on_mouse(event, x, y, flags): 
    if event == cv.CV_EVENT_LBUTTONDOWN: 
     print 'Mouse Position: '+x+', '+y 
     box.append(x, y) 

#cv2.rectangle(img, pt1, pt2, color) 
#cv2.line(img, pt1, pt2, color) 
drawing_box = False 

cv.SetMouseCallback('real image', on_mouse, 0) 
count = 0 
while(1): 
    _,img = cap.read() 
    img = cv2.blur(img, (3,3)) 

    cv2.namedWindow('real image') 
    cv2.imshow('real image', img) 

    if cv2.waitKey(0) == 27: 
     cv2.destroyAllWindows() 
     break 
print box 

任何帮助感谢！

非常感谢

约翰

Answer 1

这里是我发现的修补程序：

def on_mouse(event, x, y, flags, params): 
    if event == cv.CV_EVENT_LBUTTONDOWN: 
     print 'Start Mouse Position: '+str(x)+', '+str(y) 
     sbox = [x, y] 
     boxes.append(sbox) 
    elif event == cv.CV_EVENT_LBUTTONUP: 
     print 'End Mouse Position: '+str(x)+', '+str(y) 
     ebox = [x, y] 
     boxes.append(ebox) 

count = 0 
while(1): 
    count += 1 
    _,img = cap.read() 
    img = cv2.blur(img, (3,3)) 

    cv2.namedWindow('real image') 
    cv.SetMouseCallback('real image', on_mouse, 0) 
    cv2.imshow('real image', img) 

    if count < 50: 
     if cv2.waitKey(33) == 27: 
      cv2.destroyAllWindows() 
      break 
    elif count >= 50: 
     if cv2.waitKey(0) == 27: 
      cv2.destroyAllWindows() 
      break 
     count = 0