Table of Contents
There are many medical image segmentation models out there, but few frameworks focus on mono medical image segmentation or focus on both. This project is a general medical image segmentation model in PyTorch, and to be more precise, it is a general medical image segmentation model in PyTorch that is not only able to segment medical images, but also can be used for other tasks.
Of course, no one template will serve all projects since your needs may be different. So I'll be adding more in the near future. You may also suggest changes by forking this repo and creating a pull request or opening an issue. Thanks to all the people have contributed to expanding this template!
- 2022-04-29: support multi-modal medical image segmentation and adjacent layer.
- 2022-04-26: support multi-classes and upload inference code.
- 2022-04-24: update multi-clases metrics.
- 2022-04-07: optimized logger code. Mean the metrics.
- 2022-04-06: Added Transforms , crop data by label(over-sampling) and Generate patches mask
- 2022-04-02: Added some script. examples, save data.
- 2022-04-02: Mean the metrics
- 2022-03-30: Added 3dDataloader
- 2022-03-25: Added Configuration README
- 2022-03-24: Added License section and README.md
- 2022-03-11: Fixed Release Notes
- 2022-03-10: Initial Release
This is an example of how you may give instructions on setting up your project locally. To get a local copy up and running follow these simple example steps.
Below is an example of how you can instruct your audience on installing and setting up your app. This template doesn't rely on any external dependencies or services.
-
Clone the repo
git clone https://github.com/JohnMasoner/MedicalZoo.git
-
Install Python packages
pip install -r requirements.txt
-
Initialization Configuration. you could apply change configuration to your datasets and customize your training strategies.
python run.py [configuration] [training_model]For more information about configuration, Please Config ReadMe
- Add a blog to introduce the project
- Add More Modules to better training your model(More TODO!)
- Add 3D Modeling
- Add more data augmentation
- Multi-language Support
- Chinese
- Chinese (Traditional)
- French
See the open issues for a full list of proposed features (and known issues).
Contributions are what make the open source community such an amazing place to learn, inspire, and create. Any contributions you make are greatly appreciated.
If you have a suggestion that would make this better, please fork the repo and create a pull request. You can also simply open an issue with the tag "enhancement". Don't forget to give the project a star! Thanks again!
- Fork the Project
- Create your Feature Branch (
git checkout -b feature/AmazingFeature) - Commit your Changes (
git commit -m 'Add some AmazingFeature') - Push to the Branch (
git push origin feature/AmazingFeature) - Open a Pull Request
Distributed under the MIT License. See LICENSE.txt for more information.
Mason Ma - masoner6429@gmail.com
This project was inspired by the following projects: