requests
torch>=2.2.0
torchvision>=0.17.0
tqdm
transformers<6.0.0,>=5.1.0
pydantic<3,>=2.0
supervision
pyDeprecate<0.8,>=0.6

[cli]
jsonargparse[signatures]>=4.27.7

[kornia]
kornia<1,>=0.7

[loggers]
tensorboard>=2.13.0
protobuf<4.0.0,>=3.20.0
wandb
mlflow
clearml

[lora]
peft

[onnx]
onnx<1.20,>=1.16.0
onnxsim<0.6.0
onnx_graphsurgeon
onnxruntime
polygraphy

[plus]
rfdetr_plus<2.0.0,>=1.0.1

[tflite]
onnx2tf<2.0.0,>=1.26.0
flatbuffers>=23.5.26
onnx<1.20,>=1.16.0
tf-keras>=2.16.0
tensorflow>=2.16.0

[train]
peft
pytorch_lightning<3,>=2.6
torchmetrics[detection]>=1.2
faster-coco-eval>=1.7.2
pycocotools
scipy
albumentations<3.0.0,>=1.4.24
roboflow
rf100vl

[trt]
pycuda
onnxruntime-gpu
tensorrt>=8.6.1
polygraphy

[visual]
matplotlib
pandas
seaborn
