From f813f6dcc875901c6ba7a509c14227c2292efed4 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 13 Mar 2021 20:00:03 -0800
Subject: [PATCH] autoShape() speed profiling update (#2460)

---
 models/common.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/common.py b/models/common.py
index 7ef5762efbf3..464d639a1f0b 100644
--- a/models/common.py
+++ b/models/common.py
@@ -168,7 +168,6 @@ def forward(self, x):
 
 class autoShape(nn.Module):
     # input-robust model wrapper for passing cv2/np/PIL/torch inputs. Includes preprocessing, inference and NMS
-    img_size = 640  # inference size (pixels)
     conf = 0.25  # NMS confidence threshold
     iou = 0.45  # NMS IoU threshold
     classes = None  # (optional list) filter by class
@@ -278,7 +277,8 @@ def display(self, pprint=False, show=False, save=False, render=False, save_dir='
 
     def print(self):
         self.display(pprint=True)  # print results
-        print(f'Speed: %.1f/%.1f/%.1f ms pre-process/inference/NMS per image at shape {tuple(self.s)}' % tuple(self.t))
+        print(f'Speed: %.1fms pre-process, %.1fms inference, %.1fms NMS per image at shape {tuple(self.s)}' %
+              tuple(self.t))                      
 
     def show(self):
         self.display(show=True)  # show results