RepNet-Pytorch

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: