augment.py

import cv2
import mediapipe as mp
import time
import os

# os.chmod('D:\projects\FY_project\hand_recognition\data\how_you', 0o777)



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

mpHands = mp.solutions.hands
hands = mpHands.Hands(static_image_mode=False,
                      max_num_hands=2,
                      min_detection_confidence=0.5,
                      min_tracking_confidence=0.5)
mpDraw = mp.solutions.drawing_utils

pTime = 0
cTime = 0

image_count = 21000
cx=cy=cx_max=cy_max=0

while True:
    img = cv2.imread("D:\projects\FY_project\hand_recognition\my_data\\you4.jpg") #cap.read()
    img_copy = img[:,:]
    imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    results = hands.process(imgRGB)
    if results.multi_hand_landmarks:
        cx,cy = 100000,100000
        cx_max,cy_max = -100000,-100000
        for handLms in results.multi_hand_landmarks:
            for id, lm in enumerate(handLms.landmark):
                h, w, c = img.shape
                cx = min(cx,int(lm.x *w))
                cy = min(cy,int(lm.y *h))
                cx_max = max(cx_max,int(lm.x *w))
                cy_max = max(cy_max,int(lm.y *h))
        cv2.rectangle(img,(cx-50,cy-50),(cx_max+50,cy_max+50),(0,255,0), 1)
        if image_count>17500:
            print(f'({cx},{cy}),({cx_max},{cy_max})')
            cv2.imwrite('D:\projects\FY_project\hand_recognition\data\\you\\you_'+str(image_count)+'.jpg',img_copy[cy-50:cy_max+50,cx-50:cx_max+50])
            image_count-=1

    cTime = time.time()
    fps = 1/(cTime-pTime)
    pTime = cTime

    cv2.putText(img,str(int(fps)), (10,70), cv2.FONT_HERSHEY_PLAIN, 3, (255,0,255), 3)

    cv2.imshow("Image", img)
    k = cv2.waitKey(10)
    if k==27:
        cv2.imwrite("Image.jpg",img)
        break