Examples for image sequence/video

lucidrains / axial-attention

Implementation of Axial attention - attending to multi-dimensional data efficiently

MIT License

352 stars 30 forks source link

Examples for image sequence/video #1

Closed tcapelle closed 4 years ago

tcapelle commented 4 years ago

Hello, Do you have examples of integrating this on image sequences? I am trying to get rid of ConvLSTM's for encoding sequence of images and AxialAttention may be a good starting point. Do you have an exmaple/notebook that I could look to integrate this on my type of data? Thank you for this amazing work. Thomas

lucidrains commented 4 years ago

@tcapelle Hello Thomas! I have an example for video, which is images in a sequence, in the readme!