59 lines
1.4 KiB
TOML
59 lines
1.4 KiB
TOML
[tool.poetry]
|
|
name = "detect-it"
|
|
version = "0.1.0"
|
|
description = "Detect all the things"
|
|
authors = ["slashtechno <77907286+slashtechno@users.noreply.github.com>"]
|
|
license = "MIT"
|
|
readme = "README.md"
|
|
packages = [{include = "detect_it"}]
|
|
|
|
[tool.poetry.dependencies]
|
|
python = "^3.10"
|
|
python-dotenv = "^1.0.0"
|
|
httpx = "^0.25.0"
|
|
opencv-python = "^4.8.1.78"
|
|
ultralytics = "^8.0.190"
|
|
hjson = "^3.1.0"
|
|
numpy = "^1.23.2"
|
|
torch = [
|
|
{ version = "^2.0.0+cu118", source = "torch_cu118", markers = "extra=='cuda'" },
|
|
{ version = "^2.0.0+cpu", source = "torch_cpu", markers = "extra!='cuda'" },
|
|
]
|
|
torchaudio = [
|
|
{ version = "^2.0.0+cu118", source = "torch_cu118", markers = "extra=='cuda'" },
|
|
{ version = "^2.0.0+cpu", source = "torch_cpu", markers = "extra!='cuda'" },
|
|
]
|
|
torchvision = [
|
|
{ version = "^0.15+cu118", source = "torch_cu118", markers = "extra=='cuda'" },
|
|
{ version = "^0.15+cpu", source = "torch_cpu", markers = "extra!='cuda'" },
|
|
]
|
|
|
|
|
|
[tool.poetry.group.dev.dependencies]
|
|
black = "^23.9.1"
|
|
ruff = "^0.0.291"
|
|
ipykernel = "^6.25.2"
|
|
|
|
|
|
[[tool.poetry.source]]
|
|
name = "torch_cpu"
|
|
url = "https://download.pytorch.org/whl/cpu"
|
|
priority = "supplemental"
|
|
|
|
[[tool.poetry.source]]
|
|
name = "torch_cu118"
|
|
url = "https://download.pytorch.org/whl/cu118"
|
|
priority = "supplemental"
|
|
|
|
[tool.poetry.extras]
|
|
cuda = []
|
|
|
|
[[tool.poetry.source]]
|
|
name = "PyPI"
|
|
priority = "primary"
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|
|
|