Spaces:
Running
Running
| import norfair | |
| import numpy as np | |
| def draw( | |
| paths_drawer, | |
| track_points, | |
| frame, | |
| detections, | |
| tracked_objects, | |
| coord_transformations, | |
| fix_paths, | |
| ): | |
| if fix_paths: | |
| frame = paths_drawer.draw(frame, tracked_objects, coord_transformations) | |
| elif paths_drawer is not None: | |
| frame = paths_drawer.draw(frame, tracked_objects) | |
| if track_points == "centroid": | |
| frame = norfair.draw_points(frame, tracked_objects, draw_ids=True) | |
| elif track_points == "bbox": | |
| frame = norfair.draw_boxes(frame, tracked_objects, draw_ids=True) | |
| return frame | |
| def center(points): | |
| return [np.mean(np.array(points), axis=0)] | |