From 7f884da6fafdf4033b8039c7fc782e88375e6467 Mon Sep 17 00:00:00 2001 From: JYChen Date: Mon, 15 Aug 2022 20:09:34 +0800 Subject: [PATCH] use link for vehicle model (#6645) --- .../config/examples/infer_cfg_human_attr.yml | 4 ++++ .../examples/infer_cfg_vehicle_attr.yml | 20 ++++++++++++++++ .../examples/infer_cfg_vehicle_plate.yml | 23 +++++++++++++++++++ .../pipeline/config/infer_cfg_ppvehicle.yml | 15 ++++-------- 4 files changed, 52 insertions(+), 10 deletions(-) create mode 100644 deploy/pipeline/config/examples/infer_cfg_vehicle_attr.yml create mode 100644 deploy/pipeline/config/examples/infer_cfg_vehicle_plate.yml diff --git a/deploy/pipeline/config/examples/infer_cfg_human_attr.yml b/deploy/pipeline/config/examples/infer_cfg_human_attr.yml index b4de76b3b..ff39f978e 100644 --- a/deploy/pipeline/config/examples/infer_cfg_human_attr.yml +++ b/deploy/pipeline/config/examples/infer_cfg_human_attr.yml @@ -3,6 +3,10 @@ attr_thresh: 0.5 visual: True warmup_frame: 50 +DET: + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip + batch_size: 1 + MOT: model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_pipeline.zip tracker_config: deploy/pipeline/config/tracker_config.yml diff --git a/deploy/pipeline/config/examples/infer_cfg_vehicle_attr.yml b/deploy/pipeline/config/examples/infer_cfg_vehicle_attr.yml new file mode 100644 index 000000000..926f40d47 --- /dev/null +++ b/deploy/pipeline/config/examples/infer_cfg_vehicle_attr.yml @@ -0,0 +1,20 @@ +crop_thresh: 0.5 +visual: True +warmup_frame: 50 + +DET: + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip + batch_size: 1 + +MOT: + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip + tracker_config: deploy/pipeline/config/tracker_config.yml + batch_size: 1 + enable: True + +VEHICLE_ATTR: + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/vehicle_attribute_model.zip + batch_size: 8 + color_threshold: 0.5 + type_threshold: 0.5 + enable: True diff --git a/deploy/pipeline/config/examples/infer_cfg_vehicle_plate.yml b/deploy/pipeline/config/examples/infer_cfg_vehicle_plate.yml new file mode 100644 index 000000000..9f8d0740a --- /dev/null +++ b/deploy/pipeline/config/examples/infer_cfg_vehicle_plate.yml @@ -0,0 +1,23 @@ +crop_thresh: 0.5 +visual: True +warmup_frame: 50 + +DET: + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip + batch_size: 1 + +MOT: + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip + tracker_config: deploy/pipeline/config/tracker_config.yml + batch_size: 1 + enable: True + +VEHICLE_PLATE: + det_model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/ch_PP-OCRv3_det_infer.tar.gz + det_limit_side_len: 736 + det_limit_type: "min" + rec_model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/ch_PP-OCRv3_rec_infer.tar.gz + rec_image_shape: [3, 48, 320] + rec_batch_num: 6 + word_dict_path: deploy/pipeline/ppvehicle/rec_word_dict.txt + enable: True diff --git a/deploy/pipeline/config/infer_cfg_ppvehicle.yml b/deploy/pipeline/config/infer_cfg_ppvehicle.yml index 76dfd30d4..11880ebd5 100644 --- a/deploy/pipeline/config/infer_cfg_ppvehicle.yml +++ b/deploy/pipeline/config/infer_cfg_ppvehicle.yml @@ -3,33 +3,28 @@ visual: True warmup_frame: 50 DET: - model_dir: output_inference/mot_ppyoloe_l_36e_ppvehicle/ + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip batch_size: 1 MOT: - model_dir: output_inference/mot_ppyoloe_l_36e_ppvehicle/ + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/mot_ppyoloe_l_36e_ppvehicle.zip tracker_config: deploy/pipeline/config/tracker_config.yml batch_size: 1 enable: False VEHICLE_PLATE: - det_model_dir: output_inference/ch_PP-OCRv3_det_infer/ + det_model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/ch_PP-OCRv3_det_infer.tar.gz det_limit_side_len: 736 det_limit_type: "min" - rec_model_dir: output_inference/ch_PP-OCRv3_rec_infer/ + rec_model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/ch_PP-OCRv3_rec_infer.tar.gz rec_image_shape: [3, 48, 320] rec_batch_num: 6 word_dict_path: deploy/pipeline/ppvehicle/rec_word_dict.txt enable: False VEHICLE_ATTR: - model_dir: output_inference/vehicle_attribute_infer/ + model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/vehicle_attribute_model.zip batch_size: 8 color_threshold: 0.5 type_threshold: 0.5 enable: False - -REID: - model_dir: output_inference/vehicle_reid_model/ - batch_size: 16 - enable: False -- GitLab