Skip to content

Update train.py: avoids error on 'unique_bodyparts'#2668

Merged
MMathisLab merged 1 commit intoDeepLabCut:pytorch_dlcfrom
Saintgene-Xu-lab:pytorch-dlc-patch#2667
Jul 19, 2024
Merged

Update train.py: avoids error on 'unique_bodyparts'#2668
MMathisLab merged 1 commit intoDeepLabCut:pytorch_dlcfrom
Saintgene-Xu-lab:pytorch-dlc-patch#2667

Conversation

@Zelin2001
Copy link
Contributor

This is a possible solution that works smoothly in my environment. However, I have not figured out why it is necessary to reset keypoints[i][3] to non-negative values so I left the process alone.

@jeylau
Copy link
Contributor

jeylau commented Jul 19, 2024

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 😊

@MMathisLab MMathisLab merged commit a7c0d58 into DeepLabCut:pytorch_dlc Jul 19, 2024
@Zelin2001 Zelin2001 deleted the pytorch-dlc-patch#2667 branch July 20, 2024 02:29
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