20 lines
506 B
Python
20 lines
506 B
Python
from pathlib import Path
|
|
|
|
from dodgeballtracker.logic import DetectedObjectBox, main
|
|
import numpy as np
|
|
|
|
|
|
here = Path(__file__).parent
|
|
input_dir = here / '..' / 'data'
|
|
output_dir = here / '..' / 'outputs'
|
|
|
|
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)
|