migrated to class

dodgeballtracker/__init__.py

@@ -1,6 +1,6 @@
 from pathlib import Path
 
-from dodgeballtracker.logic import DetectedObjectBox, main
+from dodgeballtracker.logic import DetectedObjectBox, DodgeballTracker
 import numpy as np
 
 
@@ -12,8 +12,9 @@ if __name__ == "__main__":
     detected_box_objects = [
         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"
 
-    main(input_video, output_video, detected_box_objects)
+    dodgeball_tracker = DodgeballTracker(input_video, output_video, detected_box_objects)
+
+    dodgeball_tracker.main()