mit-han-lab / temporal-shift-module

[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding
https://arxiv.org/abs/1811.08383
MIT License
2.07k stars 417 forks source link

Predicting on a single video? #179

Closed balaganeshmohan closed 3 years ago

balaganeshmohan commented 3 years ago

I have trained my own model with a custom dataset that is not related to action recognition. I would like to use my best trained model to predict on a single video and get its class. Sort of like how we do usually for 2D CNN. How can I modify the test file? Has anyone managed to do it before?

cryotheta commented 2 years ago

did you find a soution?