Open taosean opened 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.
const at::Tensor &trans
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!
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 indcn_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!