From 653009ab5531df53edd7ef12bddcb89c24d4b2ca Mon Sep 17 00:00:00 2001 From: zhiboniu Date: Tue, 21 Jun 2022 06:12:48 +0000 Subject: [PATCH] update some names --- deploy/pphuman/config/infer_cfg_ppvehicle.yml | 3 +-- deploy/pphuman/datacollector.py | 6 +++--- deploy/pphuman/ppvehicle/vehicle_plate.py | 1 - 3 files changed, 4 insertions(+), 6 deletions(-) diff --git a/deploy/pphuman/config/infer_cfg_ppvehicle.yml b/deploy/pphuman/config/infer_cfg_ppvehicle.yml index 523078306..b3aeedd5c 100644 --- a/deploy/pphuman/config/infer_cfg_ppvehicle.yml +++ b/deploy/pphuman/config/infer_cfg_ppvehicle.yml @@ -9,14 +9,13 @@ DET: batch_size: 1 MOT: - model_dir: output_inference/mot_ppyoloe_s_36e_ppvehicle/ + model_dir: output_inference/mot_ppyoloe_l_36e_ppvehicle/ tracker_config: deploy/pphuman/config/tracker_config.yml batch_size: 1 basemode: "idbased" enable: False VEHICLE_PLATE: - det_algorithm: "DB" det_model_dir: output_inference/ch_PP-OCRv3_det_infer/ det_limit_side_len: 480 det_limit_type: "max" diff --git a/deploy/pphuman/datacollector.py b/deploy/pphuman/datacollector.py index bf1473814..66272b6a8 100644 --- a/deploy/pphuman/datacollector.py +++ b/deploy/pphuman/datacollector.py @@ -82,7 +82,7 @@ class DataCollector(object): kpt_res = Result.get('kpt') skeleton_action_res = Result.get('skeleton_action') reid_res = Result.get('reid') - vehicplate_res = Result.get('vehicleplate') + vehicleplate_res = Result.get('vehicleplate') rects = [] if reid_res is not None: @@ -112,8 +112,8 @@ class DataCollector(object): idx]) self.collector[ids]["qualities"].append(reid_res['qualities'][ idx]) - if vehicplate_res: - self.collector[ids]["vehicleplate"].append(vehicplate_res[ + if vehicleplate_res: + self.collector[ids]["vehicleplate"].append(vehicleplate_res[ 'plate'][idx]) def get_res(self): diff --git a/deploy/pphuman/ppvehicle/vehicle_plate.py b/deploy/pphuman/ppvehicle/vehicle_plate.py index 8d26b4fb1..e8787dfe0 100644 --- a/deploy/pphuman/ppvehicle/vehicle_plate.py +++ b/deploy/pphuman/ppvehicle/vehicle_plate.py @@ -37,7 +37,6 @@ from pphuman.pipe_utils import merge_cfg, print_arguments, argsparser class PlateDetector(object): def __init__(self, args, cfg): self.args = args - self.det_algorithm = cfg['det_algorithm'] self.pre_process_list = { 'Resize_Mult32': { 'limit_side_len': cfg['det_limit_side_len'], -- GitLab