requirements.txt

torch
torchvision
transformers
numpy
matplotlib
torchmetrics
opencv-python
datasets # huggingface dataset
git+https://github.com/facebookresearch/segment-anything.git