@@ -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: |