yaml.load -> yaml.safe_load

PRBonn · Aug 9, 2019 · 5368eed · 5368eed
1 parent 8cf091b
commit 5368eed
Show file tree

Hide file tree

Showing 15 changed files with 15 additions and 15 deletions.
diff --git a/train/auxiliary/calculate_means_stds.py b/train/auxiliary/calculate_means_stds.py
@@ -45,7 +45,7 @@ def is_image(filename):
     if(FLAGS.cfg):
       print("Opening config file %s" % FLAGS.cfg)
       with open(FLAGS.cfg, 'r') as file:
-        CFG = yaml.load(file)
+        CFG = yaml.safe_load(file)
   except Exception as e:
     print(e)
     print("Error opening yaml file.")

diff --git a/train/auxiliary/calculate_segmentation_weights.py b/train/auxiliary/calculate_segmentation_weights.py
@@ -61,7 +61,7 @@ def is_image(filename):
     if(FLAGS.cfg):
       print("Opening config file %s" % FLAGS.cfg)
       with open(FLAGS.cfg, 'r') as file:
-        CFG = yaml.load(file)
+        CFG = yaml.safe_load(file)
         FLAGS.numclasses = len(CFG["dataset"]["labels"])
   except Exception as e:
     print(e)

diff --git a/train/tasks/classification/modules/traceSaver.py b/train/tasks/classification/modules/traceSaver.py
@@ -32,7 +32,7 @@ def __init__(self, path, new_path, force_img_prop=(None, None)):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/classification/modules/user.py b/train/tasks/classification/modules/user.py
@@ -26,7 +26,7 @@ def __init__(self, path, force_img_prop=(None, None)):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/classification/modules/userCaffe2.py b/train/tasks/classification/modules/userCaffe2.py
@@ -29,7 +29,7 @@ def __init__(self, path):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/classification/modules/userPytorch.py b/train/tasks/classification/modules/userPytorch.py
@@ -26,7 +26,7 @@ def __init__(self, path):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/classification/modules/userTensorRT.py b/train/tasks/classification/modules/userTensorRT.py
@@ -32,7 +32,7 @@ def __init__(self, path, workspace, calib_dataset=None):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/classification/train.py b/train/tasks/classification/train.py
@@ -77,7 +77,7 @@
       print("Opening default data file cfg.yaml from log folder")
       f = open(FLAGS.path + '/cfg.yaml', 'r')
       configfile = FLAGS.path + '/cfg.yaml'
-    CFG = yaml.load(f)
+    CFG = yaml.safe_load(f)
   except Exception as e:
     print(e)
     print("Error opening yaml file.")

diff --git a/train/tasks/segmentation/dataset/coco/generate_gt.py b/train/tasks/segmentation/dataset/coco/generate_gt.py
@@ -47,7 +47,7 @@ def is_image(filename):
   try:
     print("Opening config file")
     with open('cfg.yaml', 'r') as file:
-      CFG = yaml.load(file)
+      CFG = yaml.safe_load(file)
   except Exception as e:
     print(e)
     print("Error opening yaml file.")

diff --git a/train/tasks/segmentation/modules/traceSaver.py b/train/tasks/segmentation/modules/traceSaver.py
@@ -32,7 +32,7 @@ def __init__(self, path, new_path, force_img_prop=(None, None)):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/segmentation/modules/user.py b/train/tasks/segmentation/modules/user.py
@@ -27,7 +27,7 @@ def __init__(self, path, force_img_prop=(None, None)):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/segmentation/modules/userCaffe2.py b/train/tasks/segmentation/modules/userCaffe2.py
@@ -30,7 +30,7 @@ def __init__(self, path):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/segmentation/modules/userPytorch.py b/train/tasks/segmentation/modules/userPytorch.py
@@ -27,7 +27,7 @@ def __init__(self, path):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/segmentation/modules/userTensorRT.py b/train/tasks/segmentation/modules/userTensorRT.py
@@ -34,7 +34,7 @@ def __init__(self, path, workspace, calib_dataset=None):
     try:
       yaml_path = self.path + "/cfg.yaml"
       print("Opening config file %s" % yaml_path)
-      self.CFG = yaml.load(open(yaml_path, 'r'))
+      self.CFG = yaml.safe_load(open(yaml_path, 'r'))
     except Exception as e:
       print(e)
       print("Error opening cfg.yaml file from trained model.")

diff --git a/train/tasks/segmentation/train.py b/train/tasks/segmentation/train.py
@@ -77,7 +77,7 @@
       print("Opening default data file cfg.yaml from log folder")
       f = open(FLAGS.path + '/cfg.yaml', 'r')
       configfile = FLAGS.path + '/cfg.yaml'
-    CFG = yaml.load(f)
+    CFG = yaml.safe_load(f)
   except Exception as e:
     print(e)
     print("Error opening yaml file.")