We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a06670d commit 57e33fbCopy full SHA for 57e33fb
test.py
@@ -68,7 +68,7 @@ def test(data,
68
model.eval()
69
is_coco = data.endswith('coco.yaml') # is COCO dataset
70
with open(data) as f:
71
- data = yaml.load(f, Loader=yaml.FullLoader) # model dict
+ data = yaml.load(f, Loader=yaml.SafeLoader) # model dict
72
check_dataset(data) # check
73
nc = 1 if single_cls else int(data['nc']) # number of classes
74
iouv = torch.linspace(0.5, 0.95, 10).to(device) # iou vector for mAP@0.5:0.95
0 commit comments