mlfpm / deepof

DeepLabCut based data analysis package including pose estimation and representation learning mediated behavior recognition
MIT License
39 stars 6 forks source link

Programatic arena definition, video use, number of videos for training #26

Closed neuraldino closed 1 year ago

neuraldino commented 1 year ago

Hi, I have few questions:

lucasmiranda42 commented 1 year ago

Dear Dino,

apologies for the delay! You caught me on holidays. There is no easy way to programmatically define the arena without the GUI at the moment, although we're working on it, as it seems that arena detection improvements are a widely requested feature.

Even though you still can't define the arena programmatically, and minimal GUI interaction is still required, we added automatic detection of polygonal arenas to the development branch (hosted in a separate repository). You can find the details in this issue.

Answering your second question, DeepOF only uses key points when training unsupervised models. Arena detection is only used for some rule-based assessments (such as climbing or interacting with the walls). These unfortunately only support top-down (hence single-camera) videos at the moment, but you could use these videos only for DeepOF, and provide the multi-camera tracks which limit occlusion anyway. Let me know if you need help in this regard!

Regarding the number of videos, the upper limit would be determined by your hardware. I would strongly suggest you to try to use them all and play with the batch size if you're having issues with GPU memory.

Best, and hope you find the package useful! I'll tag the issue as an enhancement, related to the arena detection improvements mentioned above, and let you know when updates are up.

Cheers, Lucas

lucasmiranda42 commented 1 year ago

Dear @neuraldino,

We just released a new version of the package with several new features, including easier (either fully automatic or manual) arena detection. Importantly, if accurate arenas are not important or all videos should use the same parameters, you can now copy the last manually defined arena to all remaining videos.

Best, and hope this helps! Lucas