i´m training a custom dataset at I3D_Resnet50_custom Model in Action Recognition.
If im using inferene.py to test my trained model on various Testvideos i´ll get this output:
0000/0010: Video1.MP4 is predicted to class 0
0001/0010: Video2.MP4 is predicted to class 0
Can i output the probability too (e.g. … to class 0 with 56%)?
Because i want to adjust the threshold. My Model predicts everything to my learned 2 classes, even if there´s none of the classes in the video. There it should output “none”.
Greetings and thanks for your help :),