https://videosashka.com/watch/bqeUmLCgsVw/how-to-load-a-custom-dataset-with-tf-data-tensorflow