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

add vehicle times (#6327)

上级 b0cf9783
......@@ -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
......
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册