going somewhere

dodgeballtracker/__init__.py

@@ -1,6 +1,8 @@
 from pathlib import Path
 
-from dodgeballtracker.logic import DetectedObjectBox, DodgeballTracker
+from dodgeballtracker.logic import (DetectedObjectBox,
+                                    DodgeballTracker,
+                                    get_court_perspective_transform)
 import numpy as np
 
 
@@ -13,8 +15,10 @@ if __name__ == "__main__":
         DetectedObjectBox.from_viridis(threshold) for threshold in np.arange(0.5, 1, 0.05)
     ]
     input_video = input_dir / "micro.mp4"
-    output_video = output_dir / "output.mp4"
+    output_video = output_dir / "output-lmbn_n_cuhk03_d.mp4"
 
-    dodgeball_tracker = DodgeballTracker(input_video, output_video, detected_box_objects)
+    transformation_matrix = get_court_perspective_transform()
+
+    dodgeball_tracker = DodgeballTracker(input_video, output_video, detected_box_objects, transformation_matrix)
 
     dodgeball_tracker.main()