You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Mahdi Abdollah Pour
daffe87b3c
meta learning added
|
2 years ago | |
|---|---|---|
| .. | ||
| README.md | 2 years ago | |
README.md
README.md
ByteTrack-TensorRT in Python
Install TensorRT Toolkit
Please follow the TensorRT Installation Guide and torch2trt gitrepo to install TensorRT (Version 7 recommended) and torch2trt.
Convert model
You can convert the Pytorch model “bytetrack_s_mot17” to TensorRT model by running:
cd <ByteTrack_HOME>
python3 tools/trt.py -f exps/example/mot/yolox_s_mix_det.py -c pretrained/bytetrack_s_mot17.pth.tar
Run TensorRT demo
You can use the converted model_trt.pth to run TensorRT demo with 130 FPS:
cd <ByteTrack_HOME>
python3 tools/demo_track.py video -f exps/example/mot/yolox_s_mix_det.py --trt --save_result