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: