更换文档检测模型

2024-08-27 14:42:45 +08:00
parent aea6f19951
commit 1514e09c40
2072 changed files with 254336 additions and 4967 deletions
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_calling.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_calling.yml
@@ -0,0 +1,17 @@
+crop_thresh: 0.5
+visual: True
+warmup_frame: 50
+
+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
+  batch_size: 1
+  enable: True
+
+ID_BASED_CLSACTION:
+  model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/PPHGNet_tiny_calling_halfbody.zip
+  batch_size: 8
+  threshold: 0.8
+  display_frames: 80
+  skip_frame_num: 2
+  enable: True
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_fall_down.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_fall_down.yml
@@ -0,0 +1,22 @@
+crop_thresh: 0.5
+kpt_thresh: 0.2
+visual: True
+warmup_frame: 50
+
+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
+  batch_size: 1
+  enable: True
+
+KPT:
+  model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/dark_hrnet_w32_256x192.zip
+  batch_size: 8
+
+SKELETON_ACTION:
+  model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/STGCN.zip
+  batch_size: 1
+  max_frames: 50
+  display_frames: 80
+  coord_size: [384, 512]
+  enable: True
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_fight_recognition.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_fight_recognition.yml
@@ -0,0 +1,11 @@
+visual: True
+warmup_frame: 50
+
+VIDEO_ACTION:
+  model_dir: https://videotag.bj.bcebos.com/PaddleVideo-release2.3/ppTSM_fight.zip
+  batch_size: 1
+  frame_len: 8
+  sample_freq: 7
+  short_size: 340
+  target_size: 320
+  enable: True
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_human_attr.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_human_attr.yml
@@ -0,0 +1,19 @@
+crop_thresh: 0.5
+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
+  batch_size: 1
+  enable: True
+
+ATTR:
+  model_dir:  https://bj.bcebos.com/v1/paddledet/models/pipeline/PPLCNet_x1_0_person_attribute_945_infer.zip
+  batch_size: 8
+  enable: True
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_human_mot.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_human_mot.yml
@@ -0,0 +1,9 @@
+crop_thresh: 0.5
+visual: True
+warmup_frame: 50
+
+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
+  batch_size: 1
+  enable: True
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_illegal_parking.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_illegal_parking.yml
@@ -0,0 +1,19 @@
+crop_thresh: 0.5
+visual: True
+warmup_frame: 50
+
+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
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_reid.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_reid.yml
@@ -0,0 +1,14 @@
+crop_thresh: 0.5
+visual: True
+warmup_frame: 50
+
+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
+  batch_size: 1
+  enable: True
+
+REID:
+  model_dir:  https://bj.bcebos.com/v1/paddledet/models/pipeline/reid_model.zip
+  batch_size: 16
+  enable: True
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_smoking.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_smoking.yml
@@ -0,0 +1,17 @@
+crop_thresh: 0.5
+visual: True
+warmup_frame: 50
+
+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
+  batch_size: 1
+  enable: True
+
+ID_BASED_DETACTION:
+  model_dir:  https://bj.bcebos.com/v1/paddledet/models/pipeline/ppyoloe_crn_s_80e_smoking_visdrone.zip
+  batch_size: 8
+  threshold: 0.6
+  display_frames: 80
+  skip_frame_num: 2
+  enable: True
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_vehicle_attr.yml
+++ b/paddle_detection/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
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_vehicle_plate.yml
+++ b/paddle_detection/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
--- a/paddle_detection/deploy/pipeline/config/examples/infer_cfg_vehicle_violation.yml
+++ b/paddle_detection/deploy/pipeline/config/examples/infer_cfg_vehicle_violation.yml
@@ -0,0 +1,31 @@
+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
+  skip_frame_num: -1 # preferably no more than 3
+  enable: True
+
+LANE_SEG:
+  lane_seg_config: deploy/pipeline/config/lane_seg_config.yml
+  model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/pp_lite_stdc2_bdd100k.zip
+
+VEHICLE_PRESSING:
+  enable: True
+
+VEHICLE_RETROGRADE:
+  frame_len: 8
+  sample_freq: 7
+  enable: True
+  filter_horizontal_flag: True
+  keep_right_flag: True
+  deviation: 45
+  move_scale: 0.01
+  fence_line: [570,  163, 1030,  752]   #[x1,y1,x2,y2] y2>y1.
--- a/paddle_detection/deploy/pipeline/config/infer_cfg_pphuman.yml
+++ b/paddle_detection/deploy/pipeline/config/infer_cfg_pphuman.yml
@@ -0,0 +1,63 @@
+crop_thresh: 0.5
+attr_thresh: 0.5
+kpt_thresh: 0.2
+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
+  batch_size: 1
+  skip_frame_num: -1 # preferably no more than 3
+  enable: False
+
+KPT:
+  model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/dark_hrnet_w32_256x192.zip
+  batch_size: 8
+
+ATTR:
+  model_dir:  https://bj.bcebos.com/v1/paddledet/models/pipeline/PPLCNet_x1_0_person_attribute_945_infer.zip
+  batch_size: 8
+  enable: False
+
+VIDEO_ACTION:
+  model_dir: https://videotag.bj.bcebos.com/PaddleVideo-release2.3/ppTSM_fight.zip
+  batch_size: 1
+  frame_len: 8
+  sample_freq: 7
+  short_size: 340
+  target_size: 320
+  enable: False
+
+SKELETON_ACTION:
+  model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/STGCN.zip
+  batch_size: 1
+  max_frames: 50
+  display_frames: 80
+  coord_size: [384, 512]
+  enable: False
+
+ID_BASED_DETACTION:
+  model_dir:  https://bj.bcebos.com/v1/paddledet/models/pipeline/ppyoloe_crn_s_80e_smoking_visdrone.zip
+  batch_size: 8
+  threshold: 0.6
+  display_frames: 80
+  skip_frame_num: 2
+  enable: False
+
+ID_BASED_CLSACTION:
+  model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/PPHGNet_tiny_calling_halfbody.zip
+  batch_size: 8
+  threshold: 0.8
+  display_frames: 80
+  skip_frame_num: 2
+  enable: False
+
+REID:
+  model_dir:  https://bj.bcebos.com/v1/paddledet/models/pipeline/reid_model.zip
+  batch_size: 16
+  enable: False
--- a/paddle_detection/deploy/pipeline/config/infer_cfg_ppvehicle.yml
+++ b/paddle_detection/deploy/pipeline/config/infer_cfg_ppvehicle.yml
@@ -0,0 +1,48 @@
+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
+  skip_frame_num: -1 # preferably no more than 3
+  enable: False
+
+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: False
+
+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: False
+
+LANE_SEG:
+  lane_seg_config: deploy/pipeline/config/lane_seg_config.yml
+  model_dir: https://bj.bcebos.com/v1/paddledet/models/pipeline/pp_lite_stdc2_bdd100k.zip
+
+VEHICLE_PRESSING:
+  enable: False
+
+VEHICLE_RETROGRADE:
+  frame_len: 8
+  sample_freq: 7
+  enable: False
+  filter_horizontal_flag: True
+  keep_right_flag: True
+  deviation: 23
+  move_scale: 0.01
+  fence_line: []   #[x1,y1,x2,y2] y2>y1.
--- a/paddle_detection/deploy/pipeline/config/lane_seg_config.yml
+++ b/paddle_detection/deploy/pipeline/config/lane_seg_config.yml
@@ -0,0 +1,19 @@
+type: PLSLaneseg
+
+
+PLSLaneseg:
+  run_mode: 'paddle'
+  batch_size: 1
+  device: gpu
+  min_subgraph_size: 3
+  use_dynamic_shape: False
+  trt_min_shape: [100,100]
+  trt_max_shape: [2000,3000]
+  trt_opt_shape: [512,1024]
+  trt_calib_mode: False
+  cpu_threads: 10
+  enable_mkldnn: False    #Enable to use mkldnn to speed up when using cpu.
+
+  filter_horizontal_flag: True       #Whether to filter horizontal roads
+  horizontal_filtration_degree: 23
+  horizontal_filtering_threshold: 0.25
--- a/paddle_detection/deploy/pipeline/config/tracker_config.yml
+++ b/paddle_detection/deploy/pipeline/config/tracker_config.yml
@@ -0,0 +1,55 @@
+# config of tracker for MOT SDE Detector, use 'OCSORTTracker' as default, 'JDETracker' here is just BYTETracker.
+# The tracker of MOT JDE Detector (such as FairMOT) is exported together with the model.
+# Here 'min_box_area' and 'vertical_ratio' are set for pedestrian, you can modify for other objects tracking.
+
+type: BOTSORTTracker # choose one tracker in ['JDETracker', 'OCSORTTracker', 'DeepSORTTracker','BOTSORTTracker']
+# When using for MTMCT(Multi-Target Multi-Camera Tracking), you should modify to 'DeepSORTTracker'
+
+
+# just as BYTETracker, used for FairMOT in PP-Tracking project and for ByteTrack in PP-Humanv1 project
+JDETracker:
+  use_byte: True
+  det_thresh: 0.3
+  conf_thres: 0.6
+  low_conf_thres: 0.1
+  match_thres: 0.9
+  min_box_area: 0
+  vertical_ratio: 0 # 1.6 for pedestrian
+
+
+# used for OC-SORT in PP-Humanv2 project and PP-Vehicle project
+OCSORTTracker:
+  det_thresh: 0.4
+  max_age: 30
+  min_hits: 3
+  iou_threshold: 0.3
+  delta_t: 3
+  inertia: 0.2
+  min_box_area: 0
+  vertical_ratio: 0
+  use_byte: False
+  use_angle_cost: False
+
+
+# used for DeepSORT and MTMCT in PP-Tracking project
+DeepSORTTracker:
+  input_size: [64, 192] # An unique operation to scale the sub-image of the selected detected boxes to a fixed size
+  min_box_area: 0
+  vertical_ratio: -1
+  budget: 100
+  max_age: 70
+  n_init: 3
+  metric_type: cosine
+  matching_threshold: 0.2
+  max_iou_distance: 0.9
+
+BOTSORTTracker:
+  track_high_thresh: 0.3
+  track_low_thresh: 0.2
+  new_track_thresh: 0.4
+  match_thresh: 0.7
+  track_buffer: 30
+  min_box_area: 0
+  camera_motion: False
+  cmc_method: 'sparseOptFlow' # only camera_motion is True,
+                              # sparseOptFlow | files (Vidstab GMC) | orb | ecc