68 lines
1.3 KiB
TOML
68 lines
1.3 KiB
TOML
[build-system]
|
|
requires = ["setuptools>=68", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "captchbreaker"
|
|
version = "0.1.0"
|
|
description = "验证码识别多模型系统 - 调度模型 + 多专家模型两级架构"
|
|
requires-python = ">=3.10,<3.13"
|
|
dependencies = [
|
|
"torch==2.5.1",
|
|
"torchvision==0.20.1",
|
|
"onnx>=1.14.0",
|
|
"onnxscript>=0.6.0",
|
|
"onnxruntime>=1.15.0,<1.24.0",
|
|
"pillow>=10.0.0",
|
|
"numpy>=1.24.0",
|
|
"tqdm>=4.65.0",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
server = [
|
|
"fastapi>=0.100.0",
|
|
"uvicorn>=0.23.0",
|
|
"python-multipart>=0.0.6",
|
|
]
|
|
cv = [
|
|
"opencv-python>=4.8.0",
|
|
]
|
|
dev = [
|
|
"pytest>=7.0.0",
|
|
]
|
|
|
|
[project.scripts]
|
|
captcha = "cli:main"
|
|
|
|
[tool.uv.sources]
|
|
torch = [
|
|
{ index = "pytorch-cu121", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" },
|
|
]
|
|
torchvision = [
|
|
{ index = "pytorch-cu121", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" },
|
|
]
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu121"
|
|
url = "https://download.pytorch.org/whl/cu121"
|
|
explicit = true
|
|
|
|
[tool.setuptools]
|
|
py-modules = ["cli", "config", "server"]
|
|
|
|
[tool.setuptools.packages.find]
|
|
include = [
|
|
"generators",
|
|
"generators.*",
|
|
"models",
|
|
"models.*",
|
|
"training",
|
|
"training.*",
|
|
"inference",
|
|
"inference.*",
|
|
"solvers",
|
|
"solvers.*",
|
|
"utils",
|
|
"utils.*",
|
|
]
|