adsist-cms/python-detector-worker
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

change tracking parameter

Browse source
This commit is contained in:
Siwat Sirichai 2025-01-14 22:35:22 +07:00
parent 9893358022
commit b22b547fdc
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
app.py
View file

@ -24,8 +24,7 @@ if torch.cuda.is_available():
model.to('cuda') model.to('cuda')
model.track( model.track(
persist=True, persist=True,
tracker="bytetrack.yaml", # You can choose a different tracker if desired tracker="bytetrack.yaml" # You can choose a different tracker if desired
track_kps=False
) )
# Retrieve class names from the model # Retrieve class names from the model
@ -124,7 +123,7 @@ async def detect(websocket: WebSocket):
buffer = stream['buffer'] buffer = stream['buffer']
if not buffer.empty(): if not buffer.empty():
frame = buffer.get() frame = buffer.get()
results = model.track(frame, stream=False) results = model.track(frame, stream=False) # Updated for tracking
boxes = [] boxes = []
for r in results: for r in results:
for track in r.tracks: for track in r.tracks:
@ -135,7 +134,7 @@ async def detect(websocket: WebSocket):
boxes.append({ boxes.append({
"class": class_names[cls], "class": class_names[cls],
"confidence": float(track.conf), "confidence": float(track.conf),
"track_id": track_id "track_id": track_id # Added track ID
}) })
# Broadcast to all subscribers of this URL # Broadcast to all subscribers of this URL
detection_data = { detection_data = {