Hi, I know there is a 2D RoI Alignment in TorchVision (torch.ops.torchvision.roi_align).
However, I am using 3D data (feature map with shape Batch, Channel, Depth, Height, Width).
Is there a 3D RoI Alignment available ? Or how can I modify the 2D RoI Aligment to make it work for 3D data ?
Many Thanks.