quic / sense

Enhance your application with the ability to see and interact with humans using any RGB camera.
https://20bn.com/products/datasets
MIT License
734 stars 107 forks source link

Train global logistic regression model on all available data #170

Closed corneliusboehm closed 3 years ago

corneliusboehm commented 3 years ago

Before, one logistic regression would be trained per class and only on the data of the split (test or validation) that was just submitted. Because tags are now globally defined and can be present in multiple classes, it makes more sense to have one global logistic regression model as well. Also we can use both data from training and validation, because this is just used for assisting the user with annotation.

Details

corneliusboehm commented 3 years ago

One user feedback though: When I clicked on "Assisted Labelling", I had annotated ~50 videos already and it started to process those videos on the CPU right ahead, before I had time to check "Use GPU" just below. It took a really long time to process everything on the CPU... should we display "Use GPU" before "Assisted Labelling"? I guess it would be a bit weird... Or would it make sense to activate "Use GPU" by default in case the system has a GPU? I guess there could be another UI trick to fix this. Don't think this should be a blocker for this PR though.

Good feedback. I think the automatic enabling would be the nicest option, so we should add that next.