使用?OpenCV?開發(fā)虛擬鍵盤的方法

更新時(shí)間：2021年11月27日 09:49:39 作者：woshicver

OpenCV是一個(gè)強(qiáng)大的圖像處理工具，用于機(jī)器學(xué)習(xí)、圖像處理等的跨平臺(tái)開源庫，用于開發(fā)實(shí)時(shí)計(jì)算機(jī)視覺應(yīng)用程序，本文重點(diǎn)給大家介紹使用?OpenCV?開發(fā)虛擬鍵盤的方法，感興趣的朋友一起看看吧

介紹

OpenCV 是最流行的計(jì)算機(jī)視覺任務(wù)庫，它是用于機(jī)器學(xué)習(xí)、圖像處理等的跨平臺(tái)開源庫，用于開發(fā)實(shí)時(shí)計(jì)算機(jī)視覺應(yīng)用程序。

CVzone 是一個(gè)計(jì)算機(jī)視覺包，它使用 OpenCV 和 Media Pipe 庫作為其核心，使我們易于運(yùn)行，例如手部跟蹤、人臉檢測(cè)、面部標(biāo)志檢測(cè)、姿勢(shì)估計(jì)等，以及圖像處理和其他計(jì)算機(jī)視覺相關(guān)的應(yīng)用程序。

使用 OpenCV 實(shí)現(xiàn)虛擬鍵盤

讓我們創(chuàng)建一個(gè)虛擬鍵盤。

首先，讓我們安裝所需的模塊。

pip install numpy
 
pip install opencv-python
 
pip install cvzone
 
pip install pynput

使用 OpenCV 為虛擬鍵盤導(dǎo)入庫

現(xiàn)在讓我們導(dǎo)入所需的模塊

import cv2
import cvzone
from cvzone.HandTrackingModule import HandDetector
from time import sleep
import numpy as np
from pynput.keyboard import Controller

這里我們從 cvzone.HandTrackingModule 導(dǎo)入 HandDetector 模塊，然后為了使虛擬鍵盤工作，我們需要從 pynput.keyboard 導(dǎo)入Controller。

cap = cv2.VideoCapture(0, cv2.CAP_DSHOW)
cap.set(3, 1280)
cap.set(4, 720)

現(xiàn)在讓我們從 cv2.Videocapture 獲取實(shí)時(shí)輸入

detector = HandDetector(detectionCon=0.8)
keyboard_keys = [["Q", "W", "E", "R", "T", "Y", "U", "I", "O", "P"],
                  ["A", "S", "D", "F", "G", "H", "J", "K", "L", ";"],
                  ["Z", "X", "C", "V", "B", "N", "M", ",", ".", "/"]]

final_text = ""

我們以 0.8 的檢測(cè)置信度初始化 HandDetector 并將其分配給檢測(cè)器。

然后我們根據(jù)鍵盤的布局創(chuàng)建一個(gè)列表數(shù)組，并定義一個(gè)空字符串來存儲(chǔ)鍵入的鍵。

定義繪制函數(shù)

keyboard = Controller()

def draw(img, buttonList):
    for button in buttonList:
        x, y = button.pos
        w, h = button.size
        cvzone.cornerRect(img, (button.pos[0], button.pos[1],
                                                   button.size[0],button.size[0]), 20 ,rt=0)
        cv2.rectangle(img, button.pos, (int(x + w), int(y + h)), (255, 144, 30), cv2.FILLED)
        cv2.putText(img, button.text, (x + 20, y + 65),
                    cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
    return img

初始化鍵盤控制器，并定義一個(gè)名為draw()的函數(shù)，它接受兩個(gè)參數(shù)，即圖像和按鈕列表并返回圖像。在draw()函數(shù)內(nèi)部，我們使用 cvzone 的cornerRect函數(shù)在每個(gè)鍵的角落繪制矩形邊緣。這是為了讓我們的鍵盤布局看起來更好看。就像下面的圖片。

你也可以嘗試更改不同的顏色。

class Button():
    def __init__(self, pos, text, size=[85, 85]):
        self.pos = pos
        self.size = size
        self.text = text

然后我們定義一個(gè)名為 Button() 的類，并提供位置、文本和大小作為輸入，以便我們可以按照明確定義的順序排列鍵盤按鍵。

buttonList = []
# mybutton = Button([100, 100], "Q")
for k in range(len(keyboard_keys)):
    for x, key in enumerate(keyboard_keys[k]):
        buttonList.append(Button([100 * x + 25, 100 * k + 50], key))

上面的循環(huán)將遍歷鍵盤按鍵和 Button 對(duì)象，我們?cè)谄渲薪o出位置和文本作為輸入附加在一個(gè)名為 button list 的列表中。稍后我們可以將這個(gè)列表傳遞給 draw 函數(shù)以在我們的實(shí)時(shí)框架之上進(jìn)行繪制。

使用 OpenCV 的虛擬鍵盤主程序

重要的部分來了。

while True:
    success, img = cap.read()
    img = detector.findHands(img)
    lmList, bboxInfo = detector.findPosition(img)
    img = draw(img, buttonList)  # change the draw funtion to transparent_layout for transparent keys
    if lmList:
        for button in buttonList:
            x, y = button.pos
            w, h = button.size
 
if x < lmList[8][0]<x+w and y < lmList[8][1] < y+h:
cv2.rectangle(img, button.pos, (x + w, y + h),
(0, 255, 255), cv2.FILLED)
cv2.putText(img, button.text, (x + 20, y + 65),
cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
l, _, _ = detector.findDistance(8,12, img, draw=False)
print(l)
 
if l < 25:
keyboard.press(button.text)
cv2.rectangle(img, button.pos, (x + w, y + h),
(0, 255, 0), cv2.FILLED)
cv2.putText(img, button.text, (x + 20, y + 65),
cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
final_text += button.text
sleep(0.20)
 
cv2.rectangle(img, (25,350), (700, 450),
(255, 255, 255), cv2.FILLED)
cv2.putText(img, final_text, (60, 425),
cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
 
# cv2.rectangle(img, (100,100), (200,200),
# (100, 255, 0), cv2.FILLED)
# cv2.putText(img, 'Q', (120,180), cv2.FONT_HERSHEY_PLAIN, 5,
# (0, 0, 0), 5)
 
# img = mybutton.draw(img)
cv2.imshow("output", img)
cv2.waitKey(1)

在 while 循環(huán)中，首先我們讀取實(shí)時(shí)輸入幀并將其存儲(chǔ)在一個(gè)名為img的變量中。然后我們將該圖像傳遞給*檢測(cè)器.findHands()*以便在幀中找到手。然后在該圖像中，我們需要找到檢測(cè)到的手的位置和邊界框信息。

在這里我們可以找到我們的食指和中指的頂點(diǎn)之間的距離，如果兩者之間的距離小于某個(gè)閾值，那么我們就可以輸入我們所指示的字母。

一旦我們獲得了位置，我們就會(huì)遍歷整個(gè)位置列表。從該列表中，我們找到按鈕位置和按鈕大小，然后根據(jù)明確定義的方式將其繪制在框架上。

圖 1：手地標(biāo)模型

之后，我們需要找到食指和中指的頂點(diǎn)之間的距離。在上圖中，你可以看到我們需要的最高點(diǎn)是點(diǎn) 8 和點(diǎn) 12。因此，我們需要在距離查找函數(shù)中傳遞 8, 12 以獲得它們之間的距離。

在上面的代碼中，你可以看到 detector.findDistance()，我們通過了 8、12 和圖像來查找距離，并將繪制標(biāo)志設(shè)置為 false，這樣我們就不需要兩點(diǎn)之間的任何線。

如果點(diǎn)之間的距離非常小，我們將使用 press() 函數(shù)來按下按鍵。在上面的代碼keyboard.press() 中，我們傳遞button.text以顯示按下的鍵。最后，我們?cè)阪I盤布局下方繪制一個(gè)小的白色矩形框，以顯示按下的鍵。

一旦你執(zhí)行了整個(gè)代碼，它看起來像這樣。

將食指和中指靠近特定字母的頂部后，你可以鍵入該字母。

如果你需要更自定義的鍵盤布局，我們可以使鍵盤布局透明。我們只需要添加一個(gè)透明布局函數(shù)并將*draw()函數(shù)替換為transparent_layout()*函數(shù)即可。

讓我們定義transparent_layout()函數(shù)。下面是函數(shù)的代碼，它采用與draw()函數(shù)相同的輸入。在這里，我們將 numpy 的zero_like()函數(shù)分配給名為imgNew的變量，并對(duì)其執(zhí)行所需的操作，例如獲得角矩形、為每個(gè)鍵創(chuàng)建矩形框并將文本放入框內(nèi)。之后，我們將該圖像復(fù)制到一個(gè)新變量并創(chuàng)建一個(gè)imgNew掩碼，然后我們使用 OpenCV 的*addWeighted()*函數(shù)將掩碼放置在實(shí)際圖像的頂部。因此，這使鍵盤布局透明。

自定義鍵盤

def transparent_layout(img, buttonList):
    imgNew = np.zeros_like(img, np.uint8)
    for button in buttonList:
        x, y = button.pos
        cvzone.cornerRect(imgNew, (button.pos[0], button.pos[1],
                                                   button.size[0],button.size[0]), 20 ,rt=0)
        cv2.rectangle(imgNew, button.pos, (x + button.size[0], y + button.size[1]),
                                   (255, 144, 30), cv2.FILLED)
        cv2.putText(imgNew, button.text, (x + 20, y + 65),
                    cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
        out = img.copy()
        alpaha = 0.5
        mask = imgNew.astype(bool)
        print(mask.shape)
        out[mask] = cv2.addWeighted(img, alpaha, imgNew, 1-alpaha, 0)[mask] 
        return out

一旦將while 循環(huán)中的*draw()函數(shù)替換為transparent_layout()*函數(shù)，它將如下所示。（下圖）

使用 OpenCV 的虛擬鍵盤的完整代碼

下面是完整的代碼

import cv2
import cvzone
from cvzone.HandTrackingModule import HandDetector
from time import sleep
import numpy as np
from pynput.keyboard import Controller
 
cap = cv2.VideoCapture(0, cv2.CAP_DSHOW)
cap.set(3, 1280)
cap.set(4, 720)
 
detector = HandDetector(detectionCon=0.8)
keyboard_keys = [["Q", "W", "E", "R", "T", "Y", "U", "I", "O", "P"],
                  ["A", "S", "D", "F", "G", "H", "J", "K", "L", ";"],
                  ["Z", "X", "C", "V", "B", "N", "M", ",", ".", "/"]]
 
final_text = ""
 
keyboard = Controller()
 
 
def draw(img, buttonList):
    for button in buttonList:
        x, y = button.pos
        w, h = button.size
        cvzone.cornerRect(img, (button.pos[0], button.pos[1],
                                                   button.size[0],button.size[0]), 20 ,rt=0)
        cv2.rectangle(img, button.pos, (int(x + w), int(y + h)), (255, 144, 30), cv2.FILLED)
        cv2.putText(img, button.text, (x + 20, y + 65),
                    cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
    return img
 
 
def transparent_layout(img, buttonList):
    imgNew = np.zeros_like(img, np.uint8)
    for button in buttonList:
        x, y = button.pos
        cvzone.cornerRect(imgNew, (button.pos[0], button.pos[1],
                                                   button.size[0],button.size[0]), 20 ,rt=0)
        cv2.rectangle(imgNew, button.pos, (x + button.size[0], y + button.size[1]),
                                   (255, 144, 30), cv2.FILLED)
        cv2.putText(imgNew, button.text, (x + 20, y + 65),
                    cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
 
    out = img.copy()
    alpaha = 0.5
    mask = imgNew.astype(bool)
    print(mask.shape)
    out[mask] = cv2.addWeighted(img, alpaha, imgNew, 1-alpaha, 0)[mask]
    return out
 
 
class Button():
    def __init__(self, pos, text, size=[85, 85]):
        self.pos = pos
        self.size = size
        self.text = text
 
 
buttonList = []
# mybutton = Button([100, 100], "Q")
for k in range(len(keyboard_keys)):
    for x, key in enumerate(keyboard_keys[k]):
        buttonList.append(Button([100 * x + 25, 100 * k + 50], key))
 
 
while True:
    success, img = cap.read()
    img = detector.findHands(img)
    lmList, bboxInfo = detector.findPosition(img)
    img = draw(img, buttonList)  # change the draw funtion to transparent_layout for transparent keys
 
    if lmList:
        for button in buttonList:
            x, y = button.pos
            w, h = button.size
 
            if x < lmList[8][0]<x+w and y < lmList[8][1] < y+h:
                cv2.rectangle(img, button.pos, (x + w, y + h),
                              (0, 255, 255), cv2.FILLED)
                cv2.putText(img, button.text, (x + 20, y + 65),
                            cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
                l, _, _ = detector.findDistance(8,12, img, draw=False)
                print(l)
 
                if l < 25:
                    keyboard.press(button.text)
                    cv2.rectangle(img, button.pos, (x + w, y + h),
                                  (0, 255, 0), cv2.FILLED)
                    cv2.putText(img, button.text, (x + 20, y + 65),
                                cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
                    final_text += button.text
                    sleep(0.20)
 
    cv2.rectangle(img, (25,350), (700, 450),
                  (255, 255, 255), cv2.FILLED)
    cv2.putText(img, final_text, (60, 425),
                cv2.FONT_HERSHEY_PLAIN, 4, (0, 0, 0), 4)
 
    # cv2.rectangle(img, (100,100), (200,200),
    #               (100, 255, 0), cv2.FILLED)
    # cv2.putText(img, 'Q', (120,180), cv2.FONT_HERSHEY_PLAIN, 5,
    #             (0, 0, 0), 5)
 
    # img = mybutton.draw(img)
    cv2.imshow("output", img)
    cv2.waitKey(1)