f2073333b710a340403843763ba60eb1e6699916,examples/data_process/tutorial_tf_dataset_voc.py,,,#,90

Before Change

ds = ds.repeat(n_epoch)
ds = ds.prefetch(buffer_size=2048)
ds = ds.batch(batch_size)
value = ds.make_one_shot_iterator().get_next()

sess = tf.InteractiveSession()

//// get a batch of images (after data augmentation)
_, _ = sess.run(value)  // 1st time takes time to compile
st = time.time()
im, annbyte = sess.run(value)
print("took {}s".format(time.time() - st))

ann = []
for a in annbyte:

After Change

im, annbyte = next(iter(ds))
print("took {}s".format(time.time() - st))

im = im.numpy()

ann = []
for a in annbyte:
    a = a.numpy().decode()

Instances