Our great sponsors
-
WorkOS
The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
But I also don't see why it would be necessary, since PyTorch includes everything you need to build a model that handles 3D data. You can use Conv3d or something more exotic like axial attention. Here is a 3D segmentation model that is probably better than U-Net derivatives.
At least with Pytorch, it is trivial to Frankenstein things and just borrow whatever components you want from other repos. If you do want to look at a framework, I think MMDet is probably the only one that is actually used by some researchers to publish their own work: https://github.com/open-mmlab/mmdetection