label_data.py

import cv2
from glob import glob
import numpy as np
import random
from sklearn.utils import shuffle
import pickle
import os

def imageLabels():
	images_labels = []
	images = glob('gestures/*/*.jpg')
	images.sort()
	for image in images:
		print(image)
		label = image[image.find(os.sep)+1: image.rfind(os.sep)]
		img = cv2.imread(image, 0)
		images_labels.append((np.array(img, dtype=np.uint8), int(label)))
	#print(images_labels)
	return images_labels

images_labels = imageLabels()
images_labels = shuffle(shuffle(shuffle(shuffle(images_labels))))
images, labels = zip(*images_labels)
print("Length of images_labels", len(images_labels))

#print(len(labels))
train_images = images[:int(11/12*len(images))]
print("Length of train_images", len(train_images))
with open('train_images', "wb") as f:
	pickle.dump(train_images, f)
del train_images

train_labels = labels[:int(11/12*len(labels))]
print("Length of train_labels", len(train_labels))
with open('train_labels', "wb") as f:
	pickle.dump(train_labels, f)
del train_labels

#test_images = images[int(5/6*len(images)):int(11/12*len(images))]
#print("Length of test_images", len(test_images))
#with open(r'C:\Users\bhuta\Desktop\sl\sign-language-new\test_images', "wb") as f:
#	pickle.dump(test_images, f)
#del test_images

#test_labels = labels[int(5/6*len(labels)):int(11/12*len(images))]
#print("Length of test_labels", len(test_labels))
#with open(r'C:\Users\bhuta\Desktop\sl\sign-language-new\test_labels', "wb") as f:
#	pickle.dump(test_labels, f)
#del test_labels

val_images = images[int(11/12*len(images)):]
print("Length of val_images", len(val_images))
with open('val_images', "wb") as f:
	pickle.dump(val_images, f)
del val_images

val_labels = labels[int(11/12*len(labels)):]
print("Length of val_labels", len(val_labels))
with open('val_labels', "wb") as f:
	pickle.dump(val_labels, f)
del val_labels