search

moboehle / B-cos

B-cos Networks: Alignment is All we Need for Interpretability
Other
102 stars 10 forks source link

BcosConv3D #8

Open sulaimanvesal opened 1 year ago

sulaimanvesal commented 1 year ago

Hi all,

Thanks for the great contribution. I was wondering if there is a 3D version of BcosConv3D? or if there is any limitation to scale from 2D to 3D. I would like to apply it for tumor lesion segmentation in 3D MRI.

Best, Sulaiman

moboehle commented 1 year ago

Hi Sulaiman,

this is currently not part of the code base, but to adapt a nn.Conv3d layer, just follow the same approach as for the B-cos Conv2d layer we provide (modules/bcosconv2d.py)

Best Moritz