matplotlib
numpy
pandas
pycocotools
tqdm
albumentations
Pillow
torchmetrics
pyyaml
ultralytics
seaborn

# UPDATE: This needs to be updated based on your CUDA version
--extra-index-url https://download.pytorch.org/whl/cu124
torch
torchvision
torchaudio

timm
addict
yapf