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

taicaile · Jan 18, 2021 · 1f33f38 · 1f33f38
1 parent bd2bf76
commit 1f33f38
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: