Skip to content

Conversation

@n-poulsen
Copy link
Contributor

Implemented batched inference for video analysis.

  • updated the InferenceRunner classes to support batched inference
  • added a detector_batch_size attribute to the project config.yaml (with default value 1)
  • added batch_size and detector_batch_size parameters to video_inference_superanimal
  • added batch_size and detector_batch_size parameters to analyze_videos (with default value None, which means that the batch sizes used are the ones defined in the config.yaml)
  • added tests to ensure all frames are processed for top-down and bottom-up models

@MMathisLab MMathisLab merged commit 8a294c6 into pytorch_dlc Aug 9, 2024
@n-poulsen n-poulsen deleted the niels/batched_video_analysis branch August 12, 2024 07:43
@n-poulsen n-poulsen mentioned this pull request Aug 13, 2024
2 tasks
Copy link
Member

@AlexEMG AlexEMG left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Perfect!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants