Abhiram V commited on
Commit
17751b9
1 Parent(s): 4501169

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

Browse files
Files changed (1) hide show
  1. utils/autoanchor.py +1 -1
utils/autoanchor.py CHANGED
@@ -98,7 +98,7 @@ def kmean_anchors(path='./data/coco128.yaml', n=9, img_size=640, thr=4.0, gen=10
98
 
99
  if isinstance(path, str): # *.yaml file
100
  with open(path) as f:
101
- data_dict = yaml.load(f, Loader=yaml.FullLoader) # model dict
102
  from utils.datasets import LoadImagesAndLabels
103
  dataset = LoadImagesAndLabels(data_dict['train'], augment=True, rect=True)
104
  else:
 
98
 
99
  if isinstance(path, str): # *.yaml file
100
  with open(path) as f:
101
+ data_dict = yaml.load(f, Loader=yaml.SafeLoader) # model dict
102
  from utils.datasets import LoadImagesAndLabels
103
  dataset = LoadImagesAndLabels(data_dict['train'], augment=True, rect=True)
104
  else: