From 2fb8aa1f9db3d29b521a07de4f0c17c87b2c8200 Mon Sep 17 00:00:00 2001 From: Mai Thanh Minh Date: Thu, 17 Jun 2021 03:56:16 +0700 Subject: [PATCH] `is_coco` list fix (#3646) --- 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()