Does 3D version work?

[John1231983]

The 3D version has size of BxCxDxHxW. However, your implement just simple 2D version BxCxHxW for both 3D and 2D version. I guess somthing wrong

[Hsankesara]

Hi @John1231983, the 3D version is working fine. You just need to give is_2d arg as False. Okay, I've not yet implement 3D with channels and hence the dimension is BxHxWxD. I have missed it and will fix it as soon as possible. Thanks for the observation.

[oldriver-wang]

Hi @Hsankesara ， the SpatialTransformation in Voxelmorph3d is same as 2d, does it work?