From 751e02de3e0634ec8e08eb13619c77d113e2460e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 10 Sep 2018 16:26:40 +0200 Subject: [PATCH] updates --- test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test.py b/test.py index b55c039e..5ff81729 100644 --- a/test.py +++ b/test.py @@ -80,7 +80,9 @@ for batch_i, (imgs, targets) in enumerate(dataloader): # If no annotations add number of detections as incorrect if annotations.size(0) == 0: target_cls = [] - correct.extend([0 for _ in range(len(detections))]) + #correct.extend([0 for _ in range(len(detections))]) + mAPs.append(0) + continue else: target_cls = annotations[:, 0]