0
我需要在圖像中繪製一條線,以便點擊pt1並拖動到pt2。因此,會顯示一條線,我也會得到pt1和pt2的座標。我目前使用兩個單獨的鼠標點擊以下代碼畫線拖動鼠標繪製一條線,並在opencv中獲得線端點的座標Python
import numpy as np
import cv2
def get_points(im):
# Set up data to send to mouse handler
data = {}
data['im'] = im.copy()
data['points'] = []
# Set the callback function for any mouse event
cv2.imshow("Image", im)
cv2.setMouseCallback("Image", mouse_handler, data)
cv2.waitKey(0)
# Convert array to np.array
points = np.vstack(data['points']).astype(float)
return points
def mouse_handler(event, x, y, flags, data):
if event == cv2.EVENT_LBUTTONDOWN:
cv2.circle(data['im'], (x, y), 3, (0, 0, 255), 5, 16);
cv2.imshow("Image", data['im']);
if len(data['points']) < 2: # This can be changed for more or less points
data['points'].append([x, y])
# Running the code
img = cv2.imread('image.jpg', 0)
pts = get_points(img)
cv2.line(img, (pts[0][0], pts[0][1]), (pts[1][0], pts[1][1]), (0,0,0), 2)
cv2.imshow('Image', img)
cv2.waitKey(0)
它的工作原理,但並沒有解決我的問題。我希望它從pt1拖到pt2並自己繪製一條線,而不是使用點擊獲得點,然後繪製線條。因此,例如,在離開我的當前實現下面的圖片,但我想這樣做是正確的圖像進行
感謝提前你的建議。