Fixes bug where height/width were ignored in get_random_picture.
Fixes bug where height/width were ignored in get_random_picture.

PiperOrigin-RevId: 238958272

tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets