Add compatibility for TF2

Author: EdjeElectronics

generate_tfrecord.py

@@ -14,7 +14,12 @@
 import os
 import io
 import pandas as pd
-import tensorflow as tf
+
+from tensorflow.python.framework.versions import VERSION
+if VERSION >= "2.0.0a0":
+    import tensorflow.compat.v1 as tf
+else:
+    import tensorflow as tf
 
 from PIL import Image
 from object_detection.utils import dataset_util