Space Time Recurrent Memory Network - Pytorch

Space Time Recurrent Memory Network - Pytorch (wip)

Implementation of Space Time Recurrent Memory Network, recurrent network competitive with attention-based solutions for video prediction / segmentation benchmarks, with a novel technique for keeping memory constant. It is likely this is used in Tesla's self driving stack.

Citations

@misc{nguyen2021space,
    title   = {Space Time Recurrent Memory Network},
    author  = {Hung Nguyen and Fuxin Li},
    year    = {2021},
    eprint  = {2109.06474},
    archivePrefix = {arXiv},
    primaryClass = {cs.CV}
}