import random
import os
import glob

DIR = '/Users/haija/data/pascalvoc/VOCdevkit/VOC2012/JPEGImages/'

import numpy as np
import tensorflow as tf
from PIL import  Image

image_files = glob.glob(os.path.join(DIR, '*.jpg'))
# The Python of doing things [we do below, using TF i.e. using C++]
#random.shuffle(image_files)
#images = []
#for f in image_files[:10]:
#  print(f)
#  images.append(np.array(Image.open(f)))



def my_decode(file_contents):
  return tf.io.decode_jpeg(file_contents, channels=1)

dataset = tf.data.Dataset.from_tensor_slices(image_files)

dataset = dataset.map(tf.io.read_file).map(my_decode)


iter = dataset.make_initializable_iterator() 
next = iter.get_next()    
sess.run(iter.initializer)
sess.run(next)

import IPython; IPython.embed()