Simple Track demo

https://github.com/tusen-ai/SimpleTrack

GitHub - tusen-ai/SimpleTrack

 

Run

python tools/demo.py \
    --name demo \
    --det_name cp \
    --obj_type vehicle \
    --config_path configs/waymo_configs/vc_kf_giou.yaml \
    --data_folder ./demo_data/

결과로 NPZ파일이 생성.

저장되는 정보는 다음과 같다.

frame, obj_ids, bboxes, states

frame : frame 번호(따로 저장되진 않지만, 데이터 길이에서 추출 가능)

obj_ids : tracking object ID

bboxes : 각 object에 해당하는 bbox정보 (8dim)

center x

center y

center z

height

width

length

orientation

detection score

states : birth, dead, alive

 

Data

 track = tracklet.Tracklet(self.configs, self.count, input_data.dets[index], input_data.det_types[index],

                self.frame_count, aux_info=aux_info, time_stamp=input_data.time_stamp)

count : object number to assign id

dets : bbox

det_types : object type?

frame_count : frame