return processed_images

def load_data(data_dir):
    images = []
images.append(np.random.random([1, 784, 3]))
images.append(np.random.random([1, 784, 3]))
return images
if __name__ == '__main__':
    TRAIN_DATA_DIR = '/some/dir/'
sess = tf.Session()
data_feed = DataFeed(TRAIN_DATA_DIR, sess)
train_data = data_feed.process_data()
print(train_data)
Tensorflow: using a session/graph in method