From b694e52e2db0f4cb63690b34ffb52a75a6c4f788 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 27 Sep 2019 23:46:45 +0200
Subject: [PATCH] updates

---
 detect.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/detect.py b/detect.py
index 826785ce..9416c184 100644
--- a/detect.py
+++ b/detect.py
@@ -69,6 +69,9 @@ def detect(save_txt=False, save_img=False):
             img = img.unsqueeze(0)
         pred, _ = model(img)
 
+        if opt.half:
+            pred = pred.float()
+
         for i, det in enumerate(non_max_suppression(pred, opt.conf_thres, opt.nms_thres)):  # detections per image
             if webcam:  # batch_size >= 1
                 p, s, im0 = path[i], '%g: ' % i, im0s[i]