Initial Code for Tensorflow

batch_size = 32
dirpath = os.path.dirname(__file__)
classes = os.listdir('training_data')
num_classes = len(classes)
validation_size = 0.2
img_size = 128
num_channels = 3
train_path='training_data'
data = dataUtility.read_train_sets(train_path, img_size, classes, validation_size=validation_size)
print("Number of files in Training-set:\t\t{}".format(len(data.train.labels)))
print("Number of files in Validation-set:\t{}".format(len(data.valid.labels)))