search

PingchuanMa / Temporal-Shift-Module

Unofficial implementation for paper `Temporal Shift Module for Efficient Video Understanding`
58 stars 8 forks source link

Is there a full version of the code that implements tsm?Contains input and loss #2

Open l2009312042 opened 5 years ago

PingchuanMa commented 5 years ago

I provided a demo with backbone of resnet in the repo. You can simply use it as a utility module using the code from TSN-PyTorch.

wangtz commented 5 years ago

Thanks for working on this. I noticed that the model was pre-trained on Kinetics dataset, however TSN-PyTorch (https://github.com/yjxiong/tsn-pytorch) only provides utilities for UCF 101. Would you recommend what's the best approach to generate frame list for Kinetics? Thanks!