torch>=2.0
torchvision>=0.15
timm>=1.0
pytorch-lightning>=2.0
torchmetrics>=1.0
numpy>=1.23
opencv-python-headless>=4.8
huggingface_hub>=0.20
watermark>=2.3
loguru>=0.7
albumentations>=1.3
matplotlib>=3.7
transformers>=4.30
jsonargparse[signatures]>=4.27
plotly>=5.0

[all]
tensorboard>=2.14
mlflow>=2.0
wandb>=0.15
albumentations>=1.3
opencv-python-headless>=4.8
pycocotools>=2.0
onnx>=1.14
onnxruntime>=1.16
onnxscript>=0.6

[coco]
pycocotools>=2.0

[dev]
pytest>=7.0
pytest-cov
build
twine
ruff
black
pycocotools>=2.0
tensorboard>=2.14
mlflow>=2.0
wandb>=0.15
albumentations>=1.3
opencv-python-headless>=4.8
plotly>=5.0
transformers>=4.30
onnx>=1.14
onnxruntime>=1.16
onnxscript>=0.6

[docs]
zensical
mkdocstrings[python]>=0.24

[mlflow]
mlflow>=2.0

[onnx]
onnx>=1.14
onnxruntime>=1.16
onnxscript>=0.6

[tensorboard]
tensorboard>=2.14

[wandb]
wandb>=0.15
