abdollahpour
/
MetaMOT
Watch 1
Star 0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
Meta Byte Track
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.
6 Commits
1 Branch
Tree: 10a8270e28
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '10a8270e28'
${ noResults }
MetaMOT/deploy/TensorRT/python
Mahdi Abdollah Pour daffe87b3c meta learning added 3 years ago
..
README.md meta learning added 3 years ago

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