search

PistonY / torch-toolbox

🛠 Toolbox to extend PyTorch functionalities
BSD 3-Clause "New" or "Revised" License
417 stars 56 forks source link

DCNv2 #10

Open bluesky314 opened 4 years ago

bluesky314 commented 4 years ago

How exactly is DCNv2 supported in this repo?

PistonY commented 4 years ago

Hi @bluesky314 , you could call

from torchtoolbox.nn import conv
dc = conv.DeformConv2d(modulation=True)

to use this feature.