Skip to content

Conversation

@n-poulsen
Copy link
Contributor

Addresses issue #2847 - the cropping parameter wasn't given to the video iterator correctly.

Additionally, an issue was noted when predicting pose from heatmaps on an MPS device:

  • the device for the pose array was not set, while the device for the x, y and dz arrays could be either cpu, cuda or mps
  • this could lead to issues where the assignment pose[:, :, :, 0] = x would silently fail, with only a portion of the x array being assigned to the pose array

This issue was also resolved.

@maximpavliv maximpavliv merged commit 12c3afb into pytorch_dlc Jan 27, 2025
1 check passed
@n-poulsen n-poulsen mentioned this pull request Jan 27, 2025
2 tasks
@n-poulsen n-poulsen deleted the niels/fix_video_analysis_cropping branch January 27, 2025 14:20
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.

3 participants