Integrate ByteTracker and BoT-SORT trackers (#788)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
This commit is contained in:
parent
d99e04daa1
commit
ed6c54da7a
24 changed files with 1635 additions and 19 deletions
32
ultralytics/tracker/README.md
Normal file
32
ultralytics/tracker/README.md
Normal file
|
|
@ -0,0 +1,32 @@
|
|||
## Tracker
|
||||
|
||||
### Trackers
|
||||
|
||||
- [x] ByteTracker
|
||||
- [x] BoT-SORT
|
||||
|
||||
### Usage
|
||||
|
||||
python interface:
|
||||
|
||||
```python
|
||||
from ultralytics import YOLO
|
||||
|
||||
model = YOLO("yolov8n.pt") # or a segmentation model .i.e yolov8n-seg.pt
|
||||
model.track(
|
||||
source="video/streams",
|
||||
stream=True,
|
||||
tracker="botsort.yaml/bytetrack.yaml",
|
||||
...,
|
||||
)
|
||||
```
|
||||
|
||||
cli:
|
||||
|
||||
```bash
|
||||
yolo detect track source=... tracker=...
|
||||
yolo segment track source=... tracker=...
|
||||
```
|
||||
|
||||
By default, trackers will use the configuration in `ultralytics/tracker/cfg`.
|
||||
We also support using a modified tracker config file. Please refer to the tracker config files in `ultralytics/tracker/cfg`.
|
||||
Loading…
Add table
Add a link
Reference in a new issue