| 12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- # This config represents a ReID only configuration of DeepSORT, it has two uses.
- # One is used for loading the detection results and ReID model to get tracking results;
- # Another is used for exporting the ReID model to deploy infer.
- _BASE_: [
- '../../../datasets/mot.yml',
- '../../../runtime.yml',
- '../_base_/deepsort_reader_1088x608.yml',
- ]
- EvalMOTDataset:
- !MOTImageFolder
- dataset_dir: dataset/mot
- data_root: MOT16/images/train
- keep_ori_im: True # set as True in DeepSORT
- det_weights: None
- reid_weights: https://paddledet.bj.bcebos.com/models/mot/deepsort/deepsort_resnet.pdparams
- # A ReID only configuration of DeepSORT, detector should be None.
- architecture: DeepSORT
- pretrain_weights: None
- DeepSORT:
- detector: None
- reid: ResNetEmbedding
- tracker: DeepSORTTracker
- ResNetEmbedding:
- model_name: "ResNet50"
- DeepSORTTracker:
- input_size: [64, 192]
- min_box_area: 0 # filter out too small boxes
- vertical_ratio: -1 # filter out bboxes, usuallly set 1.6 for pedestrian
- budget: 100
- max_age: 70
- n_init: 3
- metric_type: cosine
- matching_threshold: 0.2
- max_iou_distance: 0.9
- motion: KalmanFilter
|
