Highlights
Biom3d automatically configures the training of a 3D U-Net for 3D semantic segmentation.
The default configuration matches the performance of nnUNet but is much easier to use both for community users and developers. Biom3d is flexible for developers: easy to understand and easy to edit.
Code architecture of Biom3d versus code architecture of nnU-Net:
Biom3d modules | nnUNet modules |
---|
| |
Illustrations generated with pydeps
module
Disclaimer: Biom3d does not include the possibility to use 2D U-Net or 3D-Cascade U-Net or Pytorch distributed parallel computing (only Pytorch Data Parallel) yet. However, these options could easily be adapted if needed.
We target two main types of users:
- Community users, who are interested in using the basic functionalities of Biom3d: GUI or CLI, predictions with ready-to-use models or default training.
- Deep-learning developers, who are interested in more advanced features: changing default configuration, writing of new Biom3d modules, Biom3d core editing etc.
[21/11/2023] NEWS! Biom3d tutorials are now available online:
🔨 Installation
For the installation details, please check our documentation here: Installation
TL;DR: here is a single line of code to install biom3d:
pip install torch biom3d
✋ Usage
For Graphical User Interface users, please check our documentation here: GUI
For Command Line Interface users, please check our documentation here: CLI
For Deep Learning developers, the tutorials are currently being cooked stayed tuned! You can check the partial API documentation already: API
TL;DR: here is a single line of code to run biom3d on the BTCV challenge and reach the same performance as nnU-Net (no cross-validation yet):
python -m biom3d.preprocess_train\
--img_dir data/btcv/Training/img\
--msk_dir data/btcv/Training/label\
--num_classes 13\
--ct_norm
⚠ Disclaimer
Warning: This repository is still a work in progress and comes with no guarantees.
Issues
Please feel free to open an issue or send me an email if any problem with biom3d appears. But please make sure first that this problem is not referenced on the FAQ page: Frequently Asked Question
📑 Citation
If you find Biom3d useful in your research, please cite:
@misc{biom3d,
title={{Biom3d} Easy-to-use Tool for 3D Semantic Segmentation of Volumetric Images using Deep Learning},
author={Guillaume Mougeot},
howpublished = {\url{https://github.com/GuillaumeMougeot/biom3d}},
year={2023}
}
💰 Fundings and Acknowledgements
This project has been inspired by the following publication: "nnU-Net: a self-configuring method for deep learning-based biomedical image segmentation", Fabian Isensee et al, Nature Method, 2021.
This project has been supported by Oxford Brookes University and the European Regional Development Fund (FEDER). It was carried out between the laboratories of iGReD (France), Institut Pascal (France) and Plant Nuclear Envelop (UK).