Skip to content

Commit a913314

Browse files
authored
Update autoanchor.py with yaml.SafeLoader (ultralytics#1971)
1 parent 8ca6b9d commit a913314

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils/autoanchor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def print_results(k):
9898

9999
if isinstance(path, str): # *.yaml file
100100
with open(path) as f:
101-
data_dict = yaml.load(f, Loader=yaml.FullLoader) # model dict
101+
data_dict = yaml.load(f, Loader=yaml.SafeLoader) # model dict
102102
from utils.datasets import LoadImagesAndLabels
103103
dataset = LoadImagesAndLabels(data_dict['train'], augment=True, rect=True)
104104
else:

0 commit comments

Comments
 (0)