未验证 提交 fe9e983d 编写于 作者: Z zhiboniu 提交者: GitHub

add vehicle times (#6327)

上级 b0cf9783
...@@ -157,7 +157,8 @@ class PipeTimer(Times): ...@@ -157,7 +157,8 @@ class PipeTimer(Times):
'reid': Times(), 'reid': Times(),
'det_action': Times(), 'det_action': Times(),
'cls_action': Times(), 'cls_action': Times(),
'vehicle_attr': Times() 'vehicle_attr': Times(),
'vehicleplate': Times()
} }
self.img_num = 0 self.img_num = 0
......
...@@ -679,8 +679,12 @@ class PipePredictor(object): ...@@ -679,8 +679,12 @@ class PipePredictor(object):
frame_rgb, mot_res) frame_rgb, mot_res)
if self.with_vehicleplate: if self.with_vehicleplate:
if frame_id > self.warmup_frame:
self.pipe_timer.module_time['vehicleplate'].start()
platelicense = self.vehicleplate_detector.get_platelicense( platelicense = self.vehicleplate_detector.get_platelicense(
crop_input) crop_input)
if frame_id > self.warmup_frame:
self.pipe_timer.module_time['vehicleplate'].end()
self.pipeline_res.update(platelicense, 'vehicleplate') self.pipeline_res.update(platelicense, 'vehicleplate')
if self.with_human_attr: if self.with_human_attr:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册