Temporal repetition counting. This repo contains code for the model as mentioned in Counting Out Time. There might be some differences to the version of the paper. There is also a modified version that is part of my college project in model_inn.py.
I have used different loss functions to train the model becuase I got better results with them.
Also I created a synthetic dataset for this task using Blender and it seems to work well. Link
All the dataset videos are at this directory.
Sample result:
Features created using self similarity and multi headed attention respectively:
