Spatial Transformer Networks for 3D image

Hi there,

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?

Thanks!

2 Likes

I am also looking for this. Were you able to find a solution? Please share if you did. Thanks.

1 Like

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