Skip to content

Efficient connectivity-preserving loss function for training neural networks to perform instance segmentation.

License

Notifications You must be signed in to change notification settings

AllenNeuralDynamics/supervoxel-loss

Repository files navigation

Efficient Connectivity-Preserving Instance Segmentation with Supervoxel-Based Loss Function

License Code Style semantic-release: angular Interrogate Coverage Python

paper | poster

Overview

This repository implements a connectivity-preserving loss function designed to improve instance segmentation of curvilinear structures. The paradigm shift here is to evaluate segmentation quality at the “structure-level” as opposed to the voxel-level. The loss is computed by detecting supervoxels in the false positive and false negative masks during training, then assigning higher penalties to supervoxels that introduce connectivity errors.

pipeline
Figure: Visualization of loss computation, see Method section for description of each step.

Method

To do...

Installation

To use the software, in the root directory, run

pip install -e .

License

supervoxel-loss is licensed under the MIT License.

About

Efficient connectivity-preserving loss function for training neural networks to perform instance segmentation.

Resources

License

Stars

Watchers

Forks

Packages

No packages published