search

talmolab / dreem

DREEM Relates Every Entities' Motion (DREEM). Global Tracking Transformers for biological multi-object tracking.
https://dreem.sleap.ai
BSD 3-Clause "New" or "Revised" License
6 stars 1 forks source link

High level API/CLI for SLEAP integration #45

Open talmo opened 2 months ago

talmo commented 2 months ago

Provide equivalents of sleap-train and sleap-track and an equivalent high-level API for one-call training and inference.

talmo commented 1 month ago

Configure it here:

https://github.com/talmolab/biogtr/blob/f0c6f3c12e881b4947ae335d23d334b4fb9d7c89/pyproject.toml#L50-L51

In the current setup, this creates a CLI called biogtr which redirects to the biogtr.cli module and calls a function within that module named cli.

It would be implemented here:

https://github.com/talmolab/biogtr/blob/f0c6f3c12e881b4947ae335d23d334b4fb9d7c89/biogtr/cli.py

Instead, I propose dreem-train and dreem-track as two explicit CLIs instead of using subcommands.

You can use hydra since we're already set up to do that.