updates

2019-08-01 02:28:11 +02:00
parent e3d100dd34
commit f9d616de9f
1 changed files with 1 additions and 1 deletions
@@ -77,7 +77,7 @@ def detect(cfg,
        # Get detections
        img = torch.from_numpy(img).unsqueeze(0).to(device)
        pred, _ = model(img)
-        det = non_max_suppression(pred, conf_thres, nms_thres)[0]
+        det = non_max_suppression(pred.float(), conf_thres, nms_thres)[0]

        if det is not None and len(det) > 0:
            # Rescale boxes from 416 to true image size