pycroscopy / atomai

Deep and Machine Learning for Microscopy
https://atomai.readthedocs.io/
MIT License
187 stars 39 forks source link

Extending DL models to 3D data #19

Open ziatdinovmax opened 3 years ago

ziatdinovmax commented 3 years ago

The semantic segmentation models need to be extended to work with 3D data. This should be very straightforward - just introduce an option to select between 1D, 2D, and 3D cases to ConvBlock, UpsampleBlock, etc.

@aghosh92 Is this something you'd be interested in?

aghosh92 commented 3 years ago

Yes. I'll be interested in these. Thank you. I'll send you updates on these tasks sometime.

tommycwong commented 1 year ago

I have added 3D support to the individual nn blocks but not to the fcnns. I was wondering what's the best way to do it. Can I also be assigned to this issue?