Fixed a bug in eval.py that caused a crash with --display when no det…

…ections are output.
simutisernestas · Oct 9, 2018 · b627472 · b627472
1 parent e55b9b0
commit b627472
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eval.py b/eval.py
@@ -119,7 +119,7 @@ def prep_display(dets_out, img, gt, gt_masks, h, w):
         classes, scores, boxes, masks = [x[:args.top_k].cpu().numpy() for x in t]
         if classes.shape[0] == 0:
             print('Warning: No detections found.')
-            return
+            return img_numpy
 
     for j in reversed(range(min(args.top_k, classes.shape[0]))):
         x1, y1, x2, y2 = boxes[j, :]