REVA-QCAV/pyproject.toml
Laurent Fainsin d8522b0b63 chore: update deps + add torchmetrics
Former-commit-id: 10ce6b869b493a0ed42809a0f56203008177530c [formerly b601b68ef557d6f1dadb2eed43a7da13440981af]
Former-commit-id: 5b0859abf64c5bb815f6e560ce8b21b32cdef893
2022-09-02 15:56:34 +02:00

43 lines
811 B
TOML

[tool.poetry]
authors = ["Laurent Fainsin <laurent@fainsin.bzh>"]
description = "Simple neural network to detect calibration spheres in images"
name = "sphereDetect"
version = "1.0.0"
[tool.poetry.dependencies]
albumentations = "^1.2.1"
matplotlib = "^3.5.3"
numpy = "^1.23.2"
onnx = "^1.12.0"
onnxruntime = "^1.12.1"
python = ">=3.8,<3.11"
pytorch-lightning = "^1.7.4"
torch = "^1.12.1"
torchmetrics = "^0.9.3"
torchvision = "^0.13.1"
wandb = "^0.13.2"
[tool.poetry.dev-dependencies]
black = "^22.8.0"
ipykernel = "^6.15.2"
isort = "^5.10.1"
[build-system]
build-backend = "poetry.core.masonry.api"
requires = ["poetry-core>=1.0.0"]
[tool.black]
exclude = '''
/(
\.git
\.venv
)/
'''
include = '\.pyi?$'
line-length = 120
target-version = ["py310"]
[tool.isort]
multi_line_output = 3
profile = "black"