33 lines
785 B
Python
33 lines
785 B
Python
"""
|
|
Dagster project for pickleball ball tracking with 3D coordinates
|
|
"""
|
|
|
|
from dagster import Definitions
|
|
from dagster_project.assets import (
|
|
extract_video_frames,
|
|
detect_court_keypoints,
|
|
detect_ball_positions,
|
|
visualize_ball_on_court,
|
|
detect_net,
|
|
calibrate_camera_3d,
|
|
compute_ball_3d_coordinates,
|
|
create_interactive_viewer
|
|
)
|
|
from dagster_project.io_managers.json_io_manager import json_io_manager
|
|
|
|
defs = Definitions(
|
|
assets=[
|
|
extract_video_frames,
|
|
detect_court_keypoints,
|
|
detect_ball_positions,
|
|
visualize_ball_on_court,
|
|
detect_net,
|
|
calibrate_camera_3d,
|
|
compute_ball_3d_coordinates,
|
|
create_interactive_viewer
|
|
],
|
|
resources={
|
|
"json_io_manager": json_io_manager
|
|
}
|
|
)
|