Skip to main content

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.py for 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:

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

med_yolo-1.0.0.tar.gz (27.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

med_yolo-1.0.0-py3-none-any.whl (22.9 kB view details)

Uploaded Python 3

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

Hashes for med_yolo-1.0.0.tar.gz
Algorithm Hash digest
SHA256 4ee3660c86449602f6312b24462cc75163ed31f3049241341265a8db49462868
MD5 053c6708d4a70401438040bd2fcad206
BLAKE2b-256 9f2f4ec7d241384e41b18ae718e286c8f90eb29cf37942eaff9bae42da929a65

See more details on using hashes here.

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

Hashes for med_yolo-1.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 eaa6181585316310a247f03cd2c461e4bd726adc5206b9b3c991b5f0839158ce
MD5 70c74684d223838c6fc3ca803d591f56
BLAKE2b-256 8c118ac487ed297f57872b54e56c1ecb43e9697ec01fe38787ac0f9f7945c72d

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page