sgoldenlab / simba

SimBA (Simple Behavioral Analysis), a pipeline and GUI for developing supervised behavioral classifiers
https://simba-uw-tf-dev.readthedocs.io/
GNU General Public License v3.0
273 stars 137 forks source link

Dispersion of body parts when they are hidden when training classifiers #258

Open Lucas97223 opened 1 year ago

Lucas97223 commented 1 year ago

I'm using Simba for training classifiers on certain behaviors. When I generate the validation video after running the machine model, I realize that when body parts are occluded by an element of the arena or another mouse, they go all over the place. I was expecting an occluded body part to disappear, as we can decide of a Bp threshold in other parts of the pipeline (ROI analysis for example). You can see a picture below:

bp pb

In this picture, a mouse is hidden in a nest located in the bottom right corner of the arena, as you can see, the points for the body parts did not disappear but go all over the place.

I am afraid this could mess with some classifications of behaviors and was therefore wondering if we could make the body part points disappear when they are occluded based on their detection probability.

Thanks in advance.