| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- # This config is an assembled config for ByteTrack MOT, used as eval/infer mode for MOT.
- _BASE_: [
- 'bytetrack_ppyoloe.yml',
- '_base_/ppyoloe_mot_reader_640x640.yml'
- ]
- weights: output/bytetrack_ppyoloe_ppvehicle9cls/model_final
- metric: MCMOT # multi-class, `MOT` for single class
- num_classes: 9
- # pedestrian(1), rider(2), car(3), truck(4), bus(5), van(6), motorcycle(7), bicycle(8), others(9)
- TestMOTDataset:
- !MOTImageFolder
- dataset_dir: dataset/mot
- keep_ori_im: True # set True if save visualization images or video
- anno_path: dataset/mot/label_list.txt # absolute path
- ### write in label_list.txt each line:
- # pedestrian
- # rider
- # car
- # truck
- # bus
- # van
- # motorcycle
- # bicycle
- # others
- ###
- det_weights: https://paddledet.bj.bcebos.com/models/mot_ppyoloe_l_36e_ppvehicle9cls.pdparams
- depth_mult: 1.0
- width_mult: 1.0
- # Tracking requires higher quality boxes, so NMS score_threshold will be higher
- PPYOLOEHead:
- nms:
- name: MultiClassNMS
- nms_top_k: 1000
- keep_top_k: 100
- score_threshold: 0.1 # 0.01 in original detector
- nms_threshold: 0.4 # 0.6 in original detector
- # BYTETracker
- JDETracker:
- use_byte: True
- match_thres: 0.9
- conf_thres: 0.2
- low_conf_thres: 0.1
- min_box_area: 0
- vertical_ratio: 0 # only use 1.6 in MOT17 pedestrian
|
