torch==2.2.2 
torchvision==0.17.2
timm==1.0.14
albumentations==2.0.4
onnx==1.14.0
onnxruntime==1.15.1
pycocotools==2.0.7
PyYAML==6.0.1
scipy==1.13.0
gradio==5.23.2
pydantic==2.10.6
opencv-python==4.9.0.80
psutil==5.9.8
py-cpuinfo==9.0.0
safetensors==0.4.3
numpy==1.26.4
thop
flash-attn