Fix bug with writing output

tools/evaluate_lanenet_on_tusimple.py

@@ -126,7 +126,8 @@ def test_lanenet_batch(src_dir, weights_path, save_dir):
                 lane_predictions.append(predictions)
 
             image_lane_predictions[input_image_name] = lane_predictions
-        json.dump(image_lane_predictions, 'predictions.json')
+        with open('predictions.json') as predictions_file:
+        json.dump(image_lane_predictions, predictions_file)
 
     return
 

tools/evaluate_performance_tusimple.py

@@ -0,0 +1,27 @@
+import argparse
+
+
+def init_args():
+    """
+
+    :return:
+    """
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--predictions_dir', type=str, help='The source tusimple lane test data dir')
+    parser.add_argument('--labels_dir', type=str, help='The model weights path')
+
+    return parser.parse_args()
+
+
+if __name__ == '__main__':
+    """
+    test code
+    """
+    # init args
+    args = init_args()
+
+    test_lanenet_batch(
+        src_dir=args.image_dir,
+        weights_path=args.weights_path,
+        save_dir=args.save_dir
+    )