From f9d616de9ffccf04c95f2744b0f2f055740efefe Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 1 Aug 2019 02:28:11 +0200
Subject: [PATCH] updates

---
 detect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/detect.py b/detect.py
index 629725f7..b58eb353 100644
--- a/detect.py
+++ b/detect.py
@@ -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