search

oeway / pytorch-deform-conv

PyTorch implementation of Deformable Convolution
MIT License
911 stars 151 forks source link

Question about implementation #10

Open lxtGH opened 6 years ago

lxtGH commented 6 years ago

in file deform_conv.py inds = indices[:, 0]input.size(1)input.size(2)+ indices[:, 1]*input.size(2) + indices[:, 2] what does it mean ?

oeway commented 6 years ago

That's a way to match the indexes for th_flatten(input), for a better understanding, I would suggest to clone the code, and change it by adding print statement to check each variables, and understand it step by step. Good luck.