diff --git a/deploy/pphuman/config/infer_cfg_ppvehicle.yml b/deploy/pphuman/config/infer_cfg_ppvehicle.yml index 5230783064a3aa5c29f58c3b542ee5c3dc8400b6..b3aeedd5cbdcd4c3565f630404dae9872d3a43d9 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 bf147381436a95de502b15679f9841a329905a63..66272b6a8328e9713cddd95572cd0a6781dfddd4 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 8d26b4fb19a496d4d933bdbbb5e90908cb6b93da..e8787dfe0f2e02c3769391e5b75470c8f73f83f7 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'],