I know there is an official pytorch STN tutorial, but that doesn’t support 3D volumetric image data since torch.nn.functional.affine_grid doesn’t support Depth. Am I correct? Do you guys know if there is any 3D STN pytorch implementation, or will pytorch support that later on?
Hmm, I believe the 3D STN is implemented in the new 1.9.0 verstion of pytorch.
Look up the descriptions of
torch.nn.functional.affine_grid
and
torch.nn.functional.grid_sample