search

haoyanbin918 / Attention-in-Attention

Apache License 2.0
10 stars 1 forks source link

I3D network #1

Open yan-ctrl opened 1 year ago

yan-ctrl commented 1 year ago

Hello, thank you for your excellent work, can this work be used in the I3D network, which is a 3D convolutional network, would you try to test the performance of your model in a 3D network?

haoyanbin918 commented 1 year ago

I appreciate your interest in our work. It is sure that the AIA module can be used in 3D CNN networks, such as I3D, C3D, GST and so on. We are happy to provide help to you if there is any problem when using it.

yan-ctrl commented 1 year ago

Thank you for your prompt reply. Since many of the commonly used behaviour recognition networks are now I3D, I would like to try your work for I3D.

haoyanbin918 commented 1 year ago

It is easy to incorporate the modules into I3D. What you need to do is to reshape the input x in the forward function from bn, c, h, w = x.size() (2D TSN/TSM) to bn, c, t, h, w = x.size() (I3D)

yan-ctrl commented 1 year ago

OK, thank you