FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [View Raw Code]   [Original HTTPS Page]

ByteTrack/deploy/TensorRT/python at main · FoundationVision/ByteTrack · GitHub

/ ByteTrack Public

Breadcrumbs

  1. ByteTrack
  2. deploy
  3. TensorRT

python

Directory actions

More options

Directory actions

More options

Latest commit

 

History

History

Breadcrumbs

  1. ByteTrack
  2. deploy
  3. TensorRT

python

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
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
Back | FazBrowse Home | New Git URL