title: Issue while generating TFrecord
body: <p>I am trying to generate tf records file with tensorflow 2.3.0 what got this error</p>
  <pre><code>  File &quot;Tensorflow/scripts/generate_tfrecord.py&quot;, line 168, in &lt;module&gt;
      tf.app.run()
    File &quot;D:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\platform\app.py&quot;, line 40, in run
      _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
    File &quot;D:\Anaconda3\envs\tensorflow\lib\site-packages\absl\app.py&quot;, line 303, in run
      _run_main(main, args)
    File &quot;D:\Anaconda3\envs\tensorflow\lib\site-packages\absl\app.py&quot;, line 251, in _run_main
      sys.exit(main(argv))
    File &quot;Tensorflow/scripts/generate_tfrecord.py&quot;, line 158, in main
      tf_example = create_tf_example(group, path)
    File &quot;Tensorflow/scripts/generate_tfrecord.py&quot;, line 132, in create_tf_example
      classes.append(class_text_to_int(row['class']))
    File &quot;Tensorflow/scripts/generate_tfrecord.py&quot;, line 101, in class_text_to_int
      return label_map_dict[row_label]
  KeyError: 'p'
  Traceback (most recent call last):
    File &quot;Tensorflow/scripts/generate_tfrecord.py&quot;, line 168, in &lt;module&gt;
      tf.app.run()
    File &quot;D:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\platform\app.py&quot;, line 40, in run
      _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
    File &quot;D:\Anaconda3\envs\tensorflow\lib\site-packages\absl\app.py&quot;, line 303, in run
      _run_main(main, args)
    File &quot;D:\Anaconda3\envs\tensorflow\lib\site-packages\absl\app.py&quot;, line 251, in _run_main
      sys.exit(main(argv))
    File &quot;Tensorflow/scripts/generate_tfrecord.py&quot;, line 158, in main
      tf_example = create_tf_example(group, path)
    File &quot;Tensorflow/scripts/generate_tfrecord.py&quot;, line 132, in create_tf_example
      classes.append(class_text_to_int(row['class']))
    File &quot;Tensorflow/scripts/generate_tfrecord.py&quot;, line 101, in class_text_to_int
      return label_map_dict[row_label]
  KeyError: 'p'
  </code></pre>
  <p>Can you tell me what is the error
  The command which I used for generating tf records are -</p>
  <pre><code>!python {SCRIPTS_PATH + '/generate_tfrecord.py'} -x {IMAGE_PATH + '/train'} -l {ANNOTATION_PATH + '/label_map.pbtxt'} -o {ANNOTATION_PATH + '/train.record'}
  !python {SCRIPTS_PATH + '/generate_tfrecord.py'} -x{IMAGE_PATH + '/test'} -l {ANNOTATION_PATH + '/label_map.pbtxt'} -o {ANNOTATION_PATH + '/test.record'}
  </code></pre>