Fix - setting cropping parameters in `analyze_videos` for PyTorch models by n-poulsen · Pull Request #2849 · DeepLabCut/DeepLabCut
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
posearray was not set, while the device for thex,yanddzarrays could be eithercpu,cudaormps - this could lead to issues where the assignment
pose[:, :, :, 0] = xwould silently fail, with only a portion of thexarray being assigned to theposearray
This issue was also resolved.