From fe2df6dcaca73144ac7df3ff1a92202c77acc5a4 Mon Sep 17 00:00:00 2001 From: Mai Thanh Minh Date: Wed, 16 Jun 2021 23:40:25 +0700 Subject: [PATCH] fix #3644 --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index cbc97b420155..a38298da54da 100644 --- a/test.py +++ b/test.py @@ -78,7 +78,7 @@ def test(data, with open(data) as f: data = yaml.safe_load(f) check_dataset(data) # check - is_coco = data['val'].endswith('coco/val2017.txt') # COCO dataset + is_coco = type(data['val']) is str and data['val'].endswith('coco/val2017.txt') # COCO dataset nc = 1 if single_cls else int(data['nc']) # number of classes iouv = torch.linspace(0.5, 0.95, 10).to(device) # iou vector for mAP@0.5:0.95 niou = iouv.numel()