Spaces:
Sleeping
Sleeping
File size: 1,636 Bytes
f968273 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import os
import pickle
import mediapipe as mp
import cv2
import matplotlib.pyplot as plt
mpHands = mp.solutions.hands
mpDrawing = mp.solutions.drawing_utils
mpDrawingStyles = mp.solutions.drawing_styles
hands = mpHands.Hands(static_image_mode=True, min_detection_confidence=0.3)
DATADIR = '../letters_fine_tune'
data = []
labels = []
for dir in os.listdir(DATADIR):
for imgName in os.listdir(os.path.join(DATADIR, dir)):
dataAux = []
x_ = []
y_ = []
imgPath = os.path.join(DATADIR, dir, imgName)
print("Reading:", imgPath)
img = cv2.imread(imgPath)
if img is None:
print("Failed to load:", imgPath)
continue
imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
results = hands.process(imgRGB)
if results.multi_hand_landmarks:
for handLandmarks in results.multi_hand_landmarks:
for i in range(len(handLandmarks.landmark)):
x = handLandmarks.landmark[i].x
y = handLandmarks.landmark[i].y
x_.append(x)
y_.append(y)
for i in range(len(handLandmarks.landmark)):
x = handLandmarks.landmark[i].x
y = handLandmarks.landmark[i].y
dataAux.append(x - min(x_))
dataAux.append(y - min(y_))
data.append(dataAux)
labels.append(dir.strip().upper())
print(labels)
file = open('../processed_data/fineTuneData.pickle', 'wb')
pickle.dump({'data': data, 'labels': labels}, file)
file.close()
|