2022-06-27 13:28:28 +00:00
|
|
|
[tool.poetry]
|
2022-09-02 13:56:34 +00:00
|
|
|
authors = ["Laurent Fainsin <laurent@fainsin.bzh>"]
|
2022-09-12 08:58:57 +00:00
|
|
|
description = "Simple neural network to detect calibration spheres in images."
|
|
|
|
|
license = "MIT"
|
2022-09-02 13:56:34 +00:00
|
|
|
name = "sphereDetect"
|
2022-09-12 08:58:57 +00:00
|
|
|
readme = "README.md"
|
|
|
|
|
version = "2.0.0"
|
2022-06-27 13:28:28 +00:00
|
|
|
|
|
|
|
|
[tool.poetry.dependencies]
|
2022-09-02 13:56:34 +00:00
|
|
|
albumentations = "^1.2.1"
|
2022-09-07 08:43:51 +00:00
|
|
|
lightning-bolts = "^0.5.0"
|
2022-09-02 13:56:34 +00:00
|
|
|
numpy = "^1.23.2"
|
2022-09-07 08:43:51 +00:00
|
|
|
pycocotools = "^2.0.4"
|
2022-07-04 13:43:54 +00:00
|
|
|
python = ">=3.8,<3.11"
|
2022-09-02 13:56:34 +00:00
|
|
|
pytorch-lightning = "^1.7.4"
|
2022-09-07 08:43:51 +00:00
|
|
|
rich = "^12.5.1"
|
2022-08-30 08:20:32 +00:00
|
|
|
torch = "^1.12.1"
|
2022-09-02 13:56:34 +00:00
|
|
|
torchmetrics = "^0.9.3"
|
2022-08-30 08:20:32 +00:00
|
|
|
torchvision = "^0.13.1"
|
2022-09-02 13:56:34 +00:00
|
|
|
wandb = "^0.13.2"
|
2022-06-27 13:28:28 +00:00
|
|
|
|
2022-09-12 08:58:57 +00:00
|
|
|
[tool.poetry.group.notebooks]
|
|
|
|
|
optional = true
|
|
|
|
|
|
|
|
|
|
[tool.poetry.group.notebooks.dependencies]
|
2022-09-15 08:40:04 +00:00
|
|
|
ipykernel = "^6.15.3"
|
2022-09-02 14:31:50 +00:00
|
|
|
matplotlib = "^3.5.3"
|
2022-09-12 08:58:57 +00:00
|
|
|
onnx = "^1.12.0"
|
2022-09-02 14:31:50 +00:00
|
|
|
onnxruntime = "^1.12.1"
|
2022-09-05 14:04:52 +00:00
|
|
|
onnxruntime-gpu = "^1.12.1"
|
2022-09-12 08:58:57 +00:00
|
|
|
|
|
|
|
|
[tool.poetry.group.dev.dependencies]
|
|
|
|
|
Flake8-pyproject = "^1.1.0"
|
|
|
|
|
bandit = "^1.7.4"
|
|
|
|
|
black = {extras = ["jupyter"], version = "^22.8.0"}
|
|
|
|
|
flake8 = "^5.0.4"
|
|
|
|
|
flake8-docstrings = "^1.6.0"
|
|
|
|
|
isort = "^5.10.1"
|
|
|
|
|
mypy = "^0.971"
|
2022-09-02 14:31:50 +00:00
|
|
|
pre-commit = "^2.20.0"
|
2022-06-27 13:28:28 +00:00
|
|
|
|
|
|
|
|
[build-system]
|
|
|
|
|
build-backend = "poetry.core.masonry.api"
|
2022-07-04 13:43:54 +00:00
|
|
|
requires = ["poetry-core>=1.0.0"]
|
2022-06-27 13:28:28 +00:00
|
|
|
|
2022-09-12 08:58:57 +00:00
|
|
|
[tool.flake8]
|
|
|
|
|
# rules ignored
|
|
|
|
|
extend-ignore = ["W503", "D401", "D403"]
|
|
|
|
|
per-file-ignores = ["__init__.py:F401", "__init__.py:D104"]
|
|
|
|
|
# black
|
|
|
|
|
ignore = "E203"
|
|
|
|
|
max-line-length = 120
|
|
|
|
|
|
2022-06-27 13:28:28 +00:00
|
|
|
[tool.black]
|
|
|
|
|
exclude = '''
|
|
|
|
|
/(
|
|
|
|
|
\.git
|
|
|
|
|
\.venv
|
|
|
|
|
)/
|
|
|
|
|
'''
|
|
|
|
|
include = '\.pyi?$'
|
|
|
|
|
line-length = 120
|
|
|
|
|
target-version = ["py310"]
|
|
|
|
|
|
|
|
|
|
[tool.isort]
|
|
|
|
|
multi_line_output = 3
|
|
|
|
|
profile = "black"
|
2022-09-12 08:58:57 +00:00
|
|
|
|
|
|
|
|
[tool.mypy]
|
|
|
|
|
python_version = "3.10"
|
|
|
|
|
warn_return_any = true
|
|
|
|
|
warn_unused_configs = true
|
