How can I extract the last 10009 last images of the official tensorflow ImageNet split?

TensorFlow General Discussion
, ,
1

In Appendix D.1., the BYOL paper states that they took the last 10009 last images of the official TensorFlow ImageNet split for the validation set to tune various hyperparameters.
I would like to reproduce the result and thus wonder how I can extract only the last 10009 last images of the official TensorFlow ImageNet split. Thank you.

3

Hi @Joohyung_Lee, load the data set as a tfds dataset and you can extract the last 10009 images as shown in the code below

TotalRecords=ds.cardinality().numpy()
ds= ds.skip(TotalRecords - 10009).take(10009)

Thank You.