datamllab / autovideo

AutoVideo: An Automated Video Action Recognition System
https://autoedge.ai/
MIT License
318 stars 36 forks source link

how can i change the Confidence Threshold for recognising Actions ? #16

Open sdpawar84 opened 1 year ago

sdpawar84 commented 1 year ago

currently, it is working with 0.5 (Confidence Threshold) for both "normal" and "object-falling" actions. I wanted to make changes in my confidence. So how can I change the Confidence Threshold?

daochenzha commented 1 year ago

@sdpawar84 You can play with make_predictions https://github.com/datamllab/autovideo/blob/main/autovideo/utils/training_utils.py#L33