samexporter/pyproject.toml at main · vietanhdev/samexporter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "samexporter"
version = "0.5.0"
description = "Export and run SAM, MobileSAM, EfficientSAM, SAM2, and SAM3 with ONNX Runtime"
authors = [
    {name = "Viet Anh Nguyen", email = "vietanh.dev@gmail.com"},
]
readme = "README.md"
requires-python = ">=3.11"
license = "MIT"
license-files = ["LICENSE", "third_party_licenses/*"]
keywords = ["sam", "segment-anything", "onnx", "onnxruntime", "segmentation"]
classifiers = [
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Operating System :: OS Independent",
]

dependencies = [
    "onnx==1.20.1",
    "opencv-python==4.11.0.86",
    "segment-anything==1.0",
    "torch==2.10.0",
    "torchvision==0.25.0",
    "timm>=0.9.16,<2",
    "numpy>=1.26,<3",
    "onnxscript==0.6.2",
    "ftfy>=6.1,<7",
    "regex>=2023.0.0",
]

[project.urls]
"Homepage" = "https://github.com/vietanhdev/samexporter"
"Bug Tracker" = "https://github.com/vietanhdev/samexporter/issues"
"Documentation" = "https://anylearning-oss.nrl.ai/docs/samexporter"
"Models" = "https://huggingface.co/nrl-ai/samexporter-onnx-models"

[project.optional-dependencies]
# onnxsim has no Windows wheel; the source tarball exceeds Windows 260-char
# path limits.  Install with: pip install samexporter[export]
export = ["onnxsim==0.5.0"]
runtime-cpu = ["onnxruntime==1.24.2"]
runtime-gpu = ["onnxruntime-gpu>=1.24,<2"]
runtime-openvino = ["onnxruntime-openvino>=1.23,<2"]
runtime-directml = ["onnxruntime-directml>=1.23,<2"]
runtime-qnn = ["onnxruntime-qnn>=1.23,<2"]
dev = ["onnxruntime==1.24.2", "ruff", "pre-commit", "pytest"]

[tool.setuptools]
packages = [
    "samexporter",
    "samexporter.assets",
    "samexporter.mobile_encoder",
    "samexporter.sam2_configs",
]

[tool.setuptools.package-data]
"samexporter.sam2_configs" = ["*.yaml", "sam2.1/*.yaml"]
"samexporter.assets" = ["*.gz"]

[tool.pytest.ini_options]
testpaths = ["tests"]
pythonpath = ["."]

[tool.ruff]
line-length = 88
target-version = "py311"
exclude = [
    ".git", "__pycache__", "build", "dist", "*.egg-info", "sam3", "third_party",
]

[tool.ruff.lint]
select = ["E", "W", "F", "I", "UP"]
ignore = [
    "E501",  # line too long
    "E741",  # ambiguous variable name
    "UP006", "UP007",
]

[tool.ruff.lint.per-file-ignores]
"tests/*" = ["F401", "F811"]