2022-06-27 13:28:28 +00:00
|
|
|
[tool.poetry]
|
2022-09-02 13:56:34 +00:00
|
|
|
authors = ["Laurent Fainsin <laurent@fainsin.bzh>"]
|
|
|
|
|
description = "Simple neural network to detect calibration spheres in images"
|
|
|
|
|
name = "sphereDetect"
|
|
|
|
|
version = "1.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-07-08 09:07:26 +00:00
|
|
|
onnx = "^1.12.0"
|
2022-09-07 08:43:51 +00:00
|
|
|
onnxruntime-training = "^1.12.0"
|
|
|
|
|
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
|
|
|
|
|
|
|
|
[tool.poetry.dev-dependencies]
|
2022-09-07 08:43:51 +00:00
|
|
|
black = {extras = ["jupyter"], version = "^22.8.0"}
|
2022-09-02 13:56:34 +00:00
|
|
|
ipykernel = "^6.15.2"
|
2022-07-04 13:43:54 +00:00
|
|
|
isort = "^5.10.1"
|
2022-09-02 14:31:50 +00:00
|
|
|
matplotlib = "^3.5.3"
|
|
|
|
|
mypy = "^0.971"
|
|
|
|
|
onnxruntime = "^1.12.1"
|
2022-09-05 14:04:52 +00:00
|
|
|
onnxruntime-gpu = "^1.12.1"
|
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
|
|
|
|
|
|
|
|
[tool.black]
|
|
|
|
|
exclude = '''
|
|
|
|
|
/(
|
|
|
|
|
\.git
|
|
|
|
|
\.venv
|
|
|
|
|
)/
|
|
|
|
|
'''
|
|
|
|
|
include = '\.pyi?$'
|
|
|
|
|
line-length = 120
|
|
|
|
|
target-version = ["py310"]
|
|
|
|
|
|
|
|
|
|
[tool.isort]
|
|
|
|
|
multi_line_output = 3
|
|
|
|
|
profile = "black"
|
