From fe9e983daeeb49be9a4af38b80cc4967afc1690f Mon Sep 17 00:00:00 2001
From: zhiboniu <31800336+zhiboniu@users.noreply.github.com>
Date: Mon, 4 Jul 2022 19:38:44 +0800
Subject: [PATCH] add vehicle times (#6327)

---
 deploy/pipeline/pipe_utils.py | 3 ++-
 deploy/pipeline/pipeline.py   | 4 ++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/deploy/pipeline/pipe_utils.py b/deploy/pipeline/pipe_utils.py
index f9d91d644..f31b42839 100644
--- a/deploy/pipeline/pipe_utils.py
+++ b/deploy/pipeline/pipe_utils.py
@@ -157,7 +157,8 @@ class PipeTimer(Times):
             'reid': Times(),
             'det_action': Times(),
             'cls_action': Times(),
-            'vehicle_attr': Times()
+            'vehicle_attr': Times(),
+            'vehicleplate': Times()
         }
         self.img_num = 0
 
diff --git a/deploy/pipeline/pipeline.py b/deploy/pipeline/pipeline.py
index 886d1e335..9e3cd292c 100644
--- a/deploy/pipeline/pipeline.py
+++ b/deploy/pipeline/pipeline.py
@@ -679,8 +679,12 @@ class PipePredictor(object):
                     frame_rgb, mot_res)
 
                 if self.with_vehicleplate:
+                    if frame_id > self.warmup_frame:
+                        self.pipe_timer.module_time['vehicleplate'].start()
                     platelicense = self.vehicleplate_detector.get_platelicense(
                         crop_input)
+                    if frame_id > self.warmup_frame:
+                        self.pipe_timer.module_time['vehicleplate'].end()
                     self.pipeline_res.update(platelicense, 'vehicleplate')
 
                 if self.with_human_attr:
-- 
GitLab