Update autoanchor.py with yaml.SafeLoader (ultralytics#1971)

fhkiel-mlaip · Jan 18, 2021 · a913314 · a913314
1 parent 8ca6b9d
commit a913314
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/autoanchor.py b/utils/autoanchor.py
@@ -98,7 +98,7 @@ def print_results(k):
 
     if isinstance(path, str):  # *.yaml file
         with open(path) as f:
-            data_dict = yaml.load(f, Loader=yaml.FullLoader)  # model dict
+            data_dict = yaml.load(f, Loader=yaml.SafeLoader)  # model dict
         from utils.datasets import LoadImagesAndLabels
         dataset = LoadImagesAndLabels(data_dict['train'], augment=True, rect=True)
     else: