Skip to content

Pre-trained model weights - official release, version 1.2 #90

@rossbar

Description

@rossbar

Pre-trained model weights have been officially released and are now available via the deepcell access portal at https://users.deepcell.org. The first official release of models/datasets are version 1.2.

Model access

Pre-trained models and training datasets are licensed under a modified Apache license for non-commercial academic use only. Account requests to https://users.deepcell.org/ are automatically evaluated for license compliance. In some cases, particularly for non-.org and .edu domains, manual review is required, which can take several days.

Quickstart

Users can ensure they have access to the latest weights by:

  1. Updating the codebase; i.e. git pull and reinstall or follow the installation instructions
  2. Set your deepcell access token: e.g. export DEEPCELL_ACCESS_TOKEN=<token-generated-on-users.deepcell.org>
  3. Downloading the latest pre-trained model weights: python -c "import cellSAM; cellSAM.get_model();"
  4. (Optional) downloading the evaluation dataset: python -c "import cellSAM; cellSAM.download_training_data();". The dataset is not required for inference, but is provided for reproducibility.

See the documentation for more information.

Previous pre-trained weights

There have been previous model weights released to aid in review/preliminary evaluation of the work. These are no longer compatible with the latest software release and have been superseded by the v1.2 release.

  • cellsam_base.pt
  • cellsam_base_v1.1.pt

Reproducibility

The official weights can be used to reproduce the results in the publication. See the docs on reproducibility for details.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Models and DatasetsRelated to pre-trained model weights and/or published datasets

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions