talmolab / sleap

A deep learning framework for multi-animal pose tracking.
https://sleap.ai
Other
427 stars 97 forks source link

Generate labels from score crashing GUI #1740

Open dlhagger opened 5 months ago

dlhagger commented 5 months ago

Bug description

I have a large number of frames (1.2M) from a long video (24 hours) and trained on all the frames to hopefully find labeling suggestions based on the prediction scores. When I attempt to load the labels from the predictions folder, it hangs on RAM and the GUI crashes. Am I doing this correctly, is there a more efficient way to load predictions for refinement?

I expect the GUI to load the predicted labels from the last training loop into the GUI and use the labeling suggestions to generate new frame to correct predictions on.

Memory hangs during loading, and GUI becomes unstable and crashes

Your personal set up

Intel(R) Core(TM) i7-8700K CPU @ 3.70GHz 3.70 GHz 32.0 GB RAM Windows 11 Pro - Build 22631.3374 NVIDIA RTX 4060 Ti

Environment packages ``` # paste output of `pip freeze` or `conda list` here ```
Logs ``` # paste relevant logs here, if any ```

Screenshots

How to reproduce

  1. Go to '...'
  2. Click on '....'
  3. Scroll down to '....'
  4. See error
eberrigan commented 3 months ago

Hi @dlhagger,

Can you open the original labels project you used for training? You should be able to generate suggestions and get predictions on the suggested frames that way. I am not sure if there is something wrong with the predictions file. Also, just to clarify, are you trying to open multiple prediction files simultaneously?

Thanks!

Elizabeth