Skip to content

Commit 3dbab4c

Browse files
committed
Use cellpose3.toml copied from appose-cellpose for Cellpose3Detector
1 parent 8080cf4 commit 3dbab4c

1 file changed

Lines changed: 42 additions & 39 deletions

File tree

  • src/main/resources/org/mastodon/mamut/detection/cellpose
Lines changed: 42 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -1,53 +1,56 @@
11
[workspace]
2-
name = 'cellpose3'
3-
version = '0.0.1'
4-
description = 'Mastodon spot detector using Cellpose 3.'
5-
authors = ['Stefan Hahmann']
6-
channels = ['conda-forge']
7-
platforms = ['osx-arm64', 'osx-64', 'linux-64', 'win-64']
8-
9-
# Assume CUDA 12 is available when solving for platforms that use it (linux-64,
10-
# win-64). On platforms without CUDA (macOS), pixi silently ignores this at
11-
# runtime.
12-
[system-requirements]
13-
cuda = '12'
2+
authors = [
3+
"Stephane Rigaud <stephane.rigaud@imba.oeaw.ac.at>",
4+
"Gaelle Letort <gaelle.letort@pasteur.fr>",
5+
"Julie Mabon <julie.mabon@pasteur.fr>",
6+
]
7+
channels = ["conda-forge"]
8+
name = "cellpose3"
9+
platforms = ["osx-arm64", "win-64", "linux-64", "osx-64"]
10+
version = "0.1.0"
11+
12+
[tasks]
13+
test = { cmd = "python cp3.py" }
1414

1515
[dependencies]
16-
python = '==3.10'
17-
numpy = '*'
18-
pip = '*'
16+
python = ">=3.10.3,<3.13"
1917

2018
[pypi-dependencies]
21-
appose = '==0.7.2'
22-
cellpose = '==3.1.1.2'
19+
appose = ">=0.10.1, <0.12"
2320

24-
# ---------------------------------------------------------------------------
25-
# WINDOWS (GPU PyTorch by default)
26-
# ---------------------------------------------------------------------------
21+
# --- Platform-specific PyTorch ---
22+
# CUDA 12.6 for Linux
23+
[target.linux.pypi-dependencies]
24+
torch = { version = ">=2.5.1", index = "https://download.pytorch.org/whl/cu126" }
25+
torchvision = { version = ">=0.20.1", index = "https://download.pytorch.org/whl/cu126" }
2726

28-
[target.win-64.dependencies]
29-
pytorch-gpu = '*'
27+
# CUDA 12.6 for Windows
28+
[target.win.pypi-dependencies]
29+
#torch = { version = ">=2.5.1", index = "https://download.pytorch.org/whl/cu126" }
30+
#torchvision = { version = ">=0.20.1", index = "https://download.pytorch.org/whl/cu126" }
31+
torch = { version = ">=2.7.0, <2.8", index = "https://download.pytorch.org/whl/cu126" }
32+
torchvision = { version = ">=0.22, <0.23", index = "https://download.pytorch.org/whl/cu126" }
3033

31-
# ---------------------------------------------------------------------------
32-
# LINUX (GPU PyTorch by default)
33-
# ---------------------------------------------------------------------------
3434

35-
[target.linux-64.dependencies]
36-
pytorch-gpu = '*'
35+
# CPU-only for macOS (no CUDA support)
36+
[target.osx-arm64.pypi-dependencies]
37+
torch = { version = ">=2.10.0, <3", index = "https://download.pytorch.org/whl/cpu" }
38+
torchvision = { version = ">=0.25.0, <0.26", index = "https://download.pytorch.org/whl/cpu" }
3739

38-
# ---------------------------------------------------------------------------
39-
# macOS INTEL (CPU PyTorch)
40-
# ---------------------------------------------------------------------------
40+
# osx-64 (should be drop in favor of osx-arm64, but we keep it for now for users on older macs)
41+
[target.osx-64.pypi-dependencies]
42+
torch = { version = ">=2.2.2, <3", index = "https://download.pytorch.org/whl/cpu" }
43+
torchvision = { version = ">=0.17.2, <0.18", index = "https://download.pytorch.org/whl/cpu" }
4144

42-
[target.osx-64.dependencies]
43-
pytorch = '*'
45+
# --- CellPose version features ---
46+
[feature.cp3.pypi-dependencies]
47+
cellpose = ">=3, <4"
4448

45-
# ---------------------------------------------------------------------------
46-
# macOS Apple Silicon (MPS via PyTorch)
47-
# ---------------------------------------------------------------------------
48-
49-
[target.osx-arm64.dependencies]
50-
pytorch = '*'
49+
[feature.cp4.pypi-dependencies]
50+
cellpose = ">=4, <5"
5151

52+
# --- Environments ---
5253
[environments]
53-
default = { solve-group = 'default' }
54+
default = { features = ["cp3"] }
55+
cp3 = { features = ["cp3"] }
56+
cp4 = { features = ["cp4"] }

0 commit comments

Comments
 (0)