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

This commit is contained in:
Abhiram V authored and GitHub committed 2021-01-18 10:47:42 -08:00
1 parent 4501169547
commit 17751b9891
1 file changed
+1 -1
+1 -1
View File
@@ -98,7 +98,7 @@ def kmean_anchors(path='./data/coco128.yaml', n=9, img_size=640, thr=4.0, gen=10
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: