torch>=1.9.0
torchvision>=0.10.0
numpy>=1.19.0
psutil>=5.8.0
Pillow>=8.0.0
tqdm>=4.60.0
opencv-python
timm>=0.6.0

[dev]
pytest>=6.0
pytest-cov>=2.0
black>=21.0
flake8>=3.8

[full]
ultralytics>=8.0.0
timm
segmentation-models-pytorch>=0.3.0
nvidia-ml-py3>=7.352.0
matplotlib>=3.3.0
seaborn>=0.11.0
