search

kenshohara / video-classification-3d-cnn-pytorch

Video classification tools using 3D ResNet
MIT License
1.1k stars 260 forks source link

special actions #16

Open LakyTT opened 6 years ago

LakyTT commented 6 years ago

Hello ,thanks for your work !And it helps me very much . Now,I want to identify these actions, if the probability of identifying a certain action is less than a certain threshold, then determine the action as a special actions named "special", and print to the result video, where should I change the code?

Best wishes for you.

LakyTT commented 6 years ago

can you give me some suggestions？ thank you！

kenshohara commented 6 years ago

See classify.py. This code calculates scores of each segment. You can use the score and theshold it. https://github.com/kenshohara/video-classification-3d-cnn-pytorch/blob/master/classify.py