DataCollection.py

import time
import cv2
import numpy as np
import math
import os
from cvzone.HandTrackingModule import HandDetector

# Create folder if not exists
folder = ("Data/L"
)
os.makedirs(folder, exist_ok=True)

# Initialize camera
cap = cv2.VideoCapture(0)

# Initialize hand detector
detector = HandDetector(maxHands=1)

counter = 0

offset = 20
imgSize = 300

while True:
    success, img = cap.read()
    if not success:
        break

    hands, img = detector.findHands(img)

    imgWhite = np.ones((imgSize, imgSize, 3), np.uint8) * 255

    if hands:
        hand = hands[0]
        x, y, w, h = hand['bbox']

        # SAFE cropping (avoid negative index crash)
        x1 = max(0, x - offset)
        y1 = max(0, y - offset)
        x2 = x + w + offset
        y2 = y + h + offset

        imgCrop = img[y1:y2, x1:x2]

        if imgCrop.size != 0:
            aspectRatio = h / w

            if aspectRatio > 1:
                k = imgSize / h
                wCal = math.ceil(k * w)

                imgResize = cv2.resize(imgCrop, (wCal, imgSize))
                wGap = math.ceil((imgSize - wCal) / 2)

                imgWhite[:, wGap:wCal + wGap] = imgResize

            else:
                k = imgSize / w
                hCal = math.ceil(k * h)

                imgResize = cv2.resize(imgCrop, (imgSize, hCal))
                hGap = math.ceil((imgSize - hCal) / 2)

                imgWhite[hGap:hCal + hGap, :] = imgResize

            cv2.imshow("ImageCrop", imgCrop)
            cv2.imshow("ImageWhite", imgWhite)

    cv2.imshow("Image", img)

    key = cv2.waitKey(1)

    # SAVE IMAGE
    if key == ord("s") and hands:
        counter += 1
        filename = f"{folder}/Image_{time.time()}.jpg"
        cv2.imwrite(filename, imgWhite)
        print("Saved:", counter)

    # EXIT
    if key == 27:  # ESC key
        break

cap.release()
cv2.destroyAllWindows()