search

DeepLabCut / DeepLabCut-live

SDK for running DeepLabCut on a live video stream
https://elifesciences.org/articles/61909
Other
190 stars 50 forks source link

Calling undefined function while extracting multiple poses #26

Closed ehsainit closed 3 years ago

ehsainit commented 3 years ago

Hey All,

I think the get_top_values function needs to be defined in the pose.py file or imported from different file, otherwise the multi_pose_predict function won't work: https://github.com/DeepLabCut/DeepLabCut-live/blob/51ad7f4cd574d9086e095bc63de947b2c6a3c1cc/dlclive/pose.py#L83

P.s: I saw that the get_top_values function was actually defined in the main DLC, you probably forgot to copy it to this file while adapting the code :)

https://github.com/DeepLabCut/DeepLabCut/blob/e5c3384897b39e3e33ca440ffef9a7830bb02a70/deeplabcut/pose_estimation_tensorflow/nnet/predict.py#L144

Cheers,

AlexEMG commented 3 years ago

Feel free to submit a PR!