Updated dataset_creator.py
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this [failed invocation](https://github.com/tensorflow/tensorflow/pull/60086/checks?check_run_id=12228598133) of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request. Please don't use "update <file>" commit messages. These make it harder to understand at a glance at commit history what the changes are.

https://cbea.ms/git-commit/ Is this fixing a bug or something?
Updated dataset_creator.py
Added some features and code 
Contributed 
Used documentation of tensorflow