diegofernandezc commited on
Commit
2f7897e
1 Parent(s): 0ce37c1

fix drawing to include ids

Browse files
Files changed (1) hide show
  1. demo_utils/draw.py +5 -5
demo_utils/draw.py CHANGED
@@ -11,15 +11,15 @@ def draw(
11
  coord_transformations,
12
  fix_paths,
13
  ):
14
- if track_points == "centroid":
15
- norfair.draw_points(frame, tracked_objects)
16
- elif track_points == "bbox":
17
- norfair.draw_boxes(frame, tracked_objects)
18
-
19
  if fix_paths:
20
  frame = paths_drawer.draw(frame, tracked_objects, coord_transformations)
21
  elif paths_drawer is not None:
22
  frame = paths_drawer.draw(frame, tracked_objects)
 
 
 
 
 
23
 
24
  return frame
25
 
 
11
  coord_transformations,
12
  fix_paths,
13
  ):
 
 
 
 
 
14
  if fix_paths:
15
  frame = paths_drawer.draw(frame, tracked_objects, coord_transformations)
16
  elif paths_drawer is not None:
17
  frame = paths_drawer.draw(frame, tracked_objects)
18
+
19
+ if track_points == "centroid":
20
+ frame = norfair.draw_points(frame, tracked_objects, draw_ids=True)
21
+ elif track_points == "bbox":
22
+ frame = norfair.draw_boxes(frame, tracked_objects, draw_ids=True)
23
 
24
  return frame
25