Fix None bug in detect.py (#177)

2019-03-31 08:06:49 -04:00 · 2019-03-31 08:06:49 -04:00 · 6b828f184e
parent c0cacc45a1
commit 6b828f184e
1 changed files with 1 additions and 1 deletions
--- a/detect.py
+++ b/detect.py
@ -62,7 +62,7 @@ def detect(
        pred = model(img)
        detections = non_max_suppression(pred, conf_thres, nms_thres)[0]

-        if len(detections) > 0:
+        if detections is not None and len(detections) > 0:
            # Rescale boxes from 416 to true image size
            scale_coords(img_size, detections[:, :4], im0.shape).round()