Update train.py: avoids error on 'unique_bodyparts' by Zelin2001 · Pull Request #2668 · DeepLabCut/DeepLabCut
Thanks @Zelin2001, I find your solution very elegant.
I suppose it could be further simplified to:
if name == 'unique_bodyparts': keypoints[-1] = max(0, keypoints[-1])
but that's really a detail.
Thanks a lot for the PR 😊
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters