Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
utils/autoanchor.py
+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 if isinstance(path, str): # *.yaml file
with open(path) as f: 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 from utils.datasets import LoadImagesAndLabels
dataset = LoadImagesAndLabels(data_dict['train'], augment=True, rect=True) dataset = LoadImagesAndLabels(data_dict['train'], augment=True, rect=True)
else: else: