使用手势控制鼠标

时间:2019-02-17 09:24:59

标签: python opencv gesture

运行以下代码时,出现以下错误:

  

回溯(最近通话最近一次):

     

文件

中的文件“ path \ gesture_mouse.py”,第63行
   cv2.circle(img, (cx, cy), (w+h)/4, (0, 0, 255), 2)
     

TypeError:预期为整数参数,为浮点数

  [ WARN:0] terminating async callback

我尝试分别指定预定义的中心和半径。

import cv2
import numpy as np
from pynput.mouse import Button, Controller
import wx

mouse = Controller()

app = wx.App(False)

(sx, sy) = wx.GetDisplaySize()
(camx, camy) = (320, 240)

lowerBound = np.array([33, 80, 40])
upperBound = np.array([102, 255, 255])

cam = cv2.VideoCapture(0)

kernelOpen = np.ones((5, 5))

kernelClose = np.ones((20, 20))

pinchFlag = 0

while True:

    ret, img = cam.read()

    img=cv2.resize(img, (340, 220))

    # convert BGR to HSV
    imgHSV = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    # create the Mask
    mask = cv2.inRange(imgHSV, lowerBound, upperBound)
    # morphology
    maskOpen = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernelOpen)
    maskClose = cv2.morphologyEx(maskOpen, cv2.MORPH_CLOSE, kernelClose)

    maskFinal = maskClose
    conts, h = cv2.findContours(maskFinal.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)

    if len(conts) == 2:
        if pinchFlag == 1:
            pinchFlag = 0
            mouse.release(Button.left)
        x1, y1, w1, h1 = cv2.boundingRect(conts[0])
        x2, y2, w2, h2 = cv2.boundingRect(conts[1])
        cv2.rectangle(img, (x1, y1), (x1+w1, y1+h1), (255, 0, 0), 2)
        cv2.rectangle(img, (x2, y2), (x2+w2, y2+h2), (255, 0, 0), 2)
        cx1 = x1+w1/2
        cy1 = y1+h1/2
        cx2 = x2+w2/2
        cy2 = y2+h2/2
        cx = (cx1+cx2)/2
        cy = (cy1+cy2)/2
        cv2.line(img, (cx1, cy1), (cx2, cy2), (255, 0, 0), 2)
        cv2.circle(img, (cx, cy), 2, (0, 0, 255), 2)
        mouseLoc = (sx-(cx*sx/camx), cy*sy/camy)
        mouse.position = mouseLoc
        while mouse.position != mouseLoc:
            pass
    elif len(conts) == 1:
        x, y, w, h = cv2.boundingRect(conts[0])
        if pinchFlag == 0:
            pinchFlag = 1
            mouse.press(Button.left)
        cv2.rectangle(img, (x, y), (x+w, y+h), (255, 0, 0), 2)
        cx = x+w/2
        cy = y+h/2
        cv2.circle(img, (cx, cy), (w+h)/4, (0, 0, 255), 2)
        mouseLoc = (sx-(cx*sx/camx), cy*sy/camy)
        mouse.position = mouseLoc
        while mouse.position != mouseLoc:
            pass
    cv2.imshow("cam", img)
    cv2.waitKey(5)

我期望当框架中出现两个绿色对象时,它们将被刻在一个蓝色框中,并且从两个框的中心开始出现一条蓝线,其中将出现一个红色的点,表示该对象的位置。光标(鼠标)。

1 个答案:

答案 0 :(得分:0)

就像错误说的那样,该函数需要int作为输入,但是您输入浮点数。这是因为这些值是除法计算的结果。

print(type(10))      #<class 'int'>
print(type(5))       #<class 'int'>
print(type(10/5))    #<class 'float'>
print(10/5)          #2.0

解决方案是将计算值转换为整数:

cv2.circle(img, (int(cx), int(cy), int((w+h)/4), (0, 0, 255), 2)