Abhiram V
commited on
Commit
•
17751b9
1
Parent(s):
4501169
Update autoanchor.py with yaml.SafeLoader (#1971)
Browse files- 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.
|
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:
|