Tracking cells and lineage with deep learning.

pip install DeepCell-Tracking==0.6.1


DeepCell Tracking Banner

PyPI version Build Status Coverage Status

deepcell-tracking uses deep learning models from deepcell-tf within an assignment problem framework to track cells through time-lapse sequences and build cell lineages. The assignment problem is solved using the Hungarian algorithm.

Getting Started

deepcell-tracking is a Python package that can be installed with pip:

pip install deepcell-tracking

Or it can be installed from source:

git clone

cd deepcell-tracking

# install the dependencies
pip install .

How to Use

from deepcell_tracking import CellTracker

# X and y are the time-sequence data and their corresponding segmentations (labels), respectively.
# model is a deepcell-tf tracking model.
tracker = CellTracker(X, y, model)

tracker.track_cells()  # runs in place, builds tracks

# Save all tracked data and lineage files to a .trk file

# Open the track file
from deepcell_tracking.utils import load_trks

data = load_trks('./results.trk')

lineage = data['lineages']  # linage information
X = data['X']  # raw X data
y = data['y']  # tracked y data