Project-MONAI / Project-MONAI/MONAI
Versatile distance transforms (Euclidean and Geodesic)
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 8.7k
- Forks
- 1.6k
- Avg merge
- 5d 1h
- Merged PRs (30d)
- 20
Description
Is your feature request related to a problem? Please describe.
Several segmentation pipelines, and in particular interactive segementation ones, rely on some form of distance transform (e.g. Euclidean, Chamfer or Geodesic Distance Transform).
At the moment there is no off-the-shelf pytorch function to compute these.
Describe the solution you'd like
A MONAI-based implementation of distance transforms (ideally differentiable) would be fantastic. It woul accelerate uptake of such methods.
Describe alternatives you've considered
- Relying on scipy functions such as distance_transform_edt and distance_transform_cdt
- Rely on external code such as GeodisTK
Additional context
Example use cases of distance transforms in segmentaiton:
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
The issue names no repository files, tests, or entry points. Start by reviewing the linked MONAI pull request and the scipy distance_transform_edt and distance_transform_cdt alternatives, then clarify the supported Euclidean and geodesic behavior and differentiability requirements. Done means a MONAI-based implementation covers the agreed distance transforms with appropriate validation.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- computer-vision, machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 30/100