English | 简体中文
Pedestrian detection and tracking is widely used in the intelligent community, industrial inspection, transportation monitoring and so on. PP-Human has the detection and tracking module, which is fundamental to keypoint detection, attribute action recognition, etc. Users enjoy easy access to pretrained models here.
Task | Algorithm | Precision | Inference Speed(ms) | Download Link |
---|---|---|---|---|
Pedestrian Detection/ Tracking | PP-YOLOE | mAP: 56.3 MOTA: 72.0 |
Detection: 28ms Tracking:33.1ms |
Download Link |
./output_inference
.During the image input, the start command is as follows:
python deploy/pphuman/pipeline.py --config deploy/pphuman/config/infer_cfg.yml \
--image_file=test_image.jpg \
--device=gpu
python
python deploy/pphuman/pipeline.py --config deploy/pphuman/config/infer_cfg.yml \
--video_file=test_video.mp4 \
--device=gpu
There are two ways to modify the model path:
./deploy/pphuman/config/infer_cfg.yml
, you can configurate different model paths,which is proper only if you match keypoint models and action recognition models with the fields of DET
and MOT
respectively, and modify the corresponding path of each field into the expected path.--model_dir
in the command line to revise the model path:python deploy/pphuman/pipeline.py --config deploy/pphuman/config/infer_cfg.yml \
--video_file=test_video.mp4 \
--device=gpu \
--model_dir det=ppyoloe/
--do_entrance_counting \
--draw_center_traj
Note:
--do_entrance_counting
is whether to calculate flow at the gateway, and the default setting is False.--draw_center_traj
means whether to draw the track, and the default setting is False. It's worth noting that the test video of track drawing should be filmed by the still camera.
The test result is:Data source and copyright owner:Skyinfor Technology. Thanks for the provision of actual scenario data, which are only used for academic research here.
Get the pedestrian detection box of the image/ video input through object detection and multi-object tracking. The adopted model is PP-YOLOE, and for details, please refer to PP-YOLOE.
The multi-object tracking model solution is based on ByteTrack, and replace the original YOLOX with P-YOLOE as the detector,and BYTETracker as the tracker.
@article{zhang2021bytetrack,
title={ByteTrack: Multi-Object Tracking by Associating Every Detection Box},
author={Zhang, Yifu and Sun, Peize and Jiang, Yi and Yu, Dongdong and Yuan, Zehuan and Luo, Ping and Liu, Wenyu and Wang, Xinggang},
journal={arXiv preprint arXiv:2110.06864},
year={2021}
}