Skip to content

Commit

Permalink
Added import comments to custom detector example
Browse files Browse the repository at this point in the history
  • Loading branch information
mikel-brostrom committed May 25, 2023
1 parent cc478ce commit 92d5b67
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -216,8 +216,16 @@ $ python examples/yolo_nas_track.py --source 0

```python
from boxmot import DeepOCSORT
from pathlib import Path

tracker = DeepOCSORT()

tracker = DeepOCSORT(
model_weights=Path('mobilenetv2_x1_4_dukemtmcreid.pt'), # which ReID model to use, when applicable
device='cuda:0', # 'cpu', 'cuda:0', 'cuda:1', ... 'cuda:N'
fp16=True, # wether to run the ReID model with half precision or not
det_thresh=0.2 # minimum valid detection confidence
)

cap = cv.VideoCapture(0)
while True:
ret, im = cap.read()
Expand Down

0 comments on commit 92d5b67

Please sign in to comment.