Medical Image segmentation with YOLO
Project description
MED-YOLO: Medical Image Segmentation with YOLO
MED-YOLO is a powerful tool for medical image segmentation using YOLO (You Only Look Once) architecture. It provides an easy-to-use interface for segmenting various medical structures including organs, tumors, vessels, and bones.
Features
- Zero-shot medical image segmentation
- Support for multiple medical image formats (DICOM, NIfTI, TIFF)
- Pre-trained models for common medical structures
- Interactive web interface using Streamlit
- Batch processing capabilities
- Post-processing tools for medical image analysis
Installation
pip install med-yolo
Quick Start
from m_yolo import predict_yolo
# Load and process a medical image
results = predict_yolo.process_image(
image_path="path/to/your/image.dcm",
model_type="organ", # or "tumor", "vessel", "bone"
confidence=0.5
)
# Visualize results
results.show()
Web Interface
Run the interactive web interface:
med-yolo
This will start a Streamlit server where you can:
- Upload medical images
- Select segmentation models
- Adjust parameters
- View and export results
Supported Image Formats
- DICOM (.dcm)
- NIfTI (.nii, .nii.gz)
- TIFF (.tif, .tiff)
- PNG (.png)
- JPEG (.jpg, .jpeg)
Supported Segmentation Types
- Organs
- Tumors
- Blood Vessels
- Bones
- Other Medical Structures
Requirements
- Python >= 3.11
- CUDA-capable GPU (recommended)
- See
setup.pyfor full list of dependencies
Citation
If you use MED-YOLO in your research, please cite:
@software{med_yolo2024,
author = {Sumit Pandey},
title = {MED-YOLO: Medical Image Segmentation with YOLO},
year = {2024},
publisher = {GitHub},
url = {https://github.com/sumit-ai-ml/MED-YOLO}
}
License
This project is licensed under the MIT License - see the LICENSE file for details.
Contributing
Contributions are welcome! Please feel free to submit a Pull Request.
Contact
For questions and support, please open an issue on GitHub or contact:
- Author: Sumit Pandey
- Email: supa@di.ku.dk
- GitHub: sumit-ai-ml
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file med_yolo-1.0.0.tar.gz.
File metadata
- Download URL: med_yolo-1.0.0.tar.gz
- Upload date:
- Size: 27.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.12.8
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
4ee3660c86449602f6312b24462cc75163ed31f3049241341265a8db49462868
|
|
| MD5 |
053c6708d4a70401438040bd2fcad206
|
|
| BLAKE2b-256 |
9f2f4ec7d241384e41b18ae718e286c8f90eb29cf37942eaff9bae42da929a65
|
File details
Details for the file med_yolo-1.0.0-py3-none-any.whl.
File metadata
- Download URL: med_yolo-1.0.0-py3-none-any.whl
- Upload date:
- Size: 22.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.12.8
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
eaa6181585316310a247f03cd2c461e4bd726adc5206b9b3c991b5f0839158ce
|
|
| MD5 |
70c74684d223838c6fc3ca803d591f56
|
|
| BLAKE2b-256 |
8c118ac487ed297f57872b54e56c1ecb43e9697ec01fe38787ac0f9f7945c72d
|