tfds pascal voc example

tfds_pascal.py

import warnings
warnings.filterwarnings("ignore")
import tensorflow_datasets as tfds
import tensorflow as tf

import matplotlib.pyplot as plt
%matplotlib inline

# tfds works in both Eager and Graph modes
tf.compat.v1.enable_eager_execution()

dataset = tfds.load("voc2007", split=tfds.Split.TRAIN, batch_size=16)
dataset = dataset.shuffle(10)

for feature in dataset.take(1):
    print(feature['image'].shape)

plt.imshow(feature['image'][5])
plt.show()

print(feature['objects']['bbox'][5])