AlexHex7 / Non-local_pytorch

Implementation of Non-local Block.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Non-local model trained on Kinetics

tonylins opened this issue · comments

Hi, I have trained a Non-local ResNet-50 TSM model on Kinetics dataset using your code here: https://github.com/MIT-HAN-LAB/temporal-shift-module (I have stated the implementation is from your GitHub repo in my code). It achieves a good performance of 75.6% on Kinetics, which is even higher than Non-local ResNet-50 I3D as reported in the paper.

Since you do not include a License file in the repo, I want to make sure you are OK with using your code. And also you could consider adding a reference to my repo to show the results on Kinetics if it helps :).

@tonylins No, problem! Thanks for telling me this.
I have created the License file and added your repo link into my ReadMe file.

By the way, I look through your paper. The idea is interesting! Especially the combination of shifting with residual connection.