Check for the Python version and fail if using Python 2

detect.py

@@ -31,6 +31,10 @@
 from utils import draw_box
 from tqdm import tqdm
 
+if sys.version_info[0] < 3:
+    print("This is a Python 3 program. Use Python 3 or higher.")
+    sys.exit(1)
+
 #-------------------------------------------------------------------------------
 # Start the show
 #-------------------------------------------------------------------------------

export_model.py

@@ -27,6 +27,10 @@
 
 from tensorflow.python.framework import graph_util
 
+if sys.version_info[0] < 3:
+    print("This is a Python 3 program. Use Python 3 or higher.")
+    sys.exit(1)
+
 #---------------------------------------------------------------------------
 # Parse the commandline
 #---------------------------------------------------------------------------

infer.py

@@ -36,6 +36,10 @@
 from utils import str2bool, load_data_source, draw_box
 from tqdm import tqdm
 
+if sys.version_info[0] < 3:
+    print("This is a Python 3 program. Use Python 3 or higher.")
+    sys.exit(1)
+
 #-------------------------------------------------------------------------------
 def sample_generator(samples, image_size, batch_size):
     image_size = (image_size.w, image_size.h)

process_dataset.py

@@ -32,6 +32,10 @@
 from utils import load_data_source, str2bool, draw_box
 from tqdm import tqdm
 
+if sys.version_info[0] < 3:
+    print("This is a Python 3 program. Use Python 3 or higher.")
+    sys.exit(1)
+
 #-------------------------------------------------------------------------------
 def annotate(data_dir, samples, colors, sample_name):
     """

train.py

@@ -35,6 +35,10 @@
 from utils import *
 from tqdm import tqdm
 
+if sys.version_info[0] < 3:
+    print("This is a Python 3 program. Use Python 3 or higher.")
+    sys.exit(1)
+
 #-------------------------------------------------------------------------------
 def compute_lr(lr_values, lr_boundaries):
     with tf.variable_scope('learning_rate'):