Skip to content

Commit e619ba6

Browse files
committed
remove fixed frames for preprocessing
1 parent 65b65c3 commit e619ba6

File tree

1 file changed

+0
-11
lines changed

1 file changed

+0
-11
lines changed

element_moseq/moseq_train.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -822,16 +822,6 @@ def make_compute(
822822

823823
# Generate overlay video for this specific recording (skip if already exists)
824824
if not overlay_video_path.exists():
825-
# Calculate frames for 1 minute of video
826-
frame_rate = fps_lookup.get(
827-
video_id, 30.0
828-
) # TODO: Default to 30fps if not found
829-
frames_for_dur = int(frame_rate * 6)
830-
831-
logger.info(
832-
f"Processing video {video_id}: {frame_rate}fps -> {frames_for_dur} frames for 1min"
833-
)
834-
835825
overlay_keypoints_on_video(
836826
video_path=video_file_path.as_posix(),
837827
coordinates=coordinates[
@@ -840,7 +830,6 @@ def make_compute(
840830
skeleton=kpms_dj_config_dict["skeleton"],
841831
bodyparts=list(use_bodyparts),
842832
output_path=overlay_video_path.as_posix(),
843-
frames=range(frames_for_dur),
844833
)
845834
logger.info(f"Generated overlay video: {overlay_video_path}")
846835
else:

0 commit comments

Comments
 (0)