Migrated to nix for dependency managment

This commit is contained in:
didier committed 2024-11-04 18:50:34 -05:00
1 parent e9da24b54d
commit 8005b38b67
11 files changed
+206 -23920

No files matched your search

+28
View File
@@ -0,0 +1,28 @@
from pathlib import Path
from dodgeballtracker.logic import (DetectedObjectBox,
DodgeballTracker,
get_court_perspective_transform)
import numpy as np
import logging
here = Path(__file__).parent
input_dir = here / '..' / 'data'
output_dir = here / '..' / 'outputs'
if __name__ == "__main__":
logging.basicConfig(filename='tracker.log', level=logging.DEBUG)
detected_box_objects = [
DetectedObjectBox.from_viridis(threshold) for threshold in np.arange(0.5, 1, 0.05)
]
input_video = input_dir / "small.mp4"
output_video = output_dir / "foo.mp4"
transformation_matrix = get_court_perspective_transform()
dodgeball_tracker = DodgeballTracker(input_video, output_video, detected_box_objects, transformation_matrix)
dodgeball_tracker.main()