You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.1 KiB
50 lines
1.1 KiB
One-Prompt Medical Image Segmentation
|
|
=====================================
|
|
|
|
Welcome to the documentation for One-Prompt Medical Image Segmentation.
|
|
|
|
.. toctree::
|
|
:maxdepth: 2
|
|
:caption: Contents:
|
|
|
|
installation
|
|
quickstart
|
|
api/index
|
|
examples/index
|
|
|
|
Overview
|
|
--------
|
|
|
|
One-Prompt Medical Image Segmentation is a deep learning framework for
|
|
medical image segmentation that uses a single prompt to segment various
|
|
types of medical images.
|
|
|
|
Features
|
|
--------
|
|
|
|
* **One-Prompt Learning**: Segment medical images using a single example
|
|
* **Multi-Dataset Support**: Works with ISIC, REFUGE, Polyp datasets
|
|
* **Flexible Architecture**: Based on SAM with custom modifications
|
|
* **Easy Integration**: Simple API for training and inference
|
|
|
|
Quick Start
|
|
-----------
|
|
|
|
.. code-block:: python
|
|
|
|
from oneprompt_seg.core import get_network
|
|
from oneprompt_seg.data import CombinedPolypDataset
|
|
|
|
# Load model
|
|
model = get_network(args, 'oneprompt')
|
|
|
|
# Create dataset
|
|
dataset = CombinedPolypDataset(args, data_path)
|
|
|
|
Indices and tables
|
|
==================
|
|
|
|
* :ref:`genindex`
|
|
* :ref:`modindex`
|
|
* :ref:`search`
|