diff --git a/utils/autoanchor.py b/utils/autoanchor.py index c00f0382ff71..5dba9f1ea22f 100644 --- 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: