Skip to content

Utility for loading CellMap data for machine learning training, utilizing PyTorch, TensorStore, Xarray, and PyDantic.

License

Notifications You must be signed in to change notification settings

janelia-cellmap/cellmap-data

Repository files navigation

CellMap logo

CellMap-Data

PyPI Build Docs GitHub License Python Version from PEP 621 TOML tests black mypy codecov

Utility for loading CellMap data for machine learning training, utilizing PyTorch, Xarray, TensorStore, and PyDantic.

You can select classes to load to construct targets separately from the labels you want to predict. This allows you to train a model to predict a subset of labels, while still using all labels to construct the target from true negatives as well as true positives.

Installation

micromamba create -n cellmap -y -c conda-forge -c pytorch python=3.11
micromamba activate cellmap
git clone https://github.com/janelia-cellmap/cellmap-data.git
cd cellmap-data
pip install -e .

About

Utility for loading CellMap data for machine learning training, utilizing PyTorch, TensorStore, Xarray, and PyDantic.

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages