lucasjinreal / DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
BSD 3-Clause "New" or "Revised" License
614 stars 125 forks source link

What's the meaning of 'const at::Tensor &trans' in dcn_v2_psroi_pooling_cuda_forward? #21

Open taosean opened 3 years ago

taosean commented 3 years ago

Hi @jinfagang, thank you for your repo.

Now I'm considering writing a 3D version of dcn_psroi_pooling. However, when I was reading the code, I don't understand what the argument const at::Tensor &trans means in dcn_v2_psroi_pooling_cuda_forward.

https://github.com/jinfagang/DCNv2_latest/blob/5957f9dc22e1679ac57b7813450d66a740fb4e61/src/cuda/dcn_v2_psroi_pooling_cuda.cu#L272-L282

Is this the offset of each grid? If not, what is the use of this argument and where can I find the offset argument?

Thank you!