requirements.txt

torch==1.13.1
torchvision==0.14.1
accelerate==0.17.0
editdistance==0.6.2
einops==0.6.0
fairscale==0.4.13
fire==0.5.0
geffnet==1.0.2
git+https://github.com/facebookresearch/detectron2.git@5aeb252b194b93dc2879b4ac34bc51a31b5aee13
git+https://github.com/openai/CLIP.git@a9b1bf5
gradio==3.20.1
huggingface-hub==0.12.1
opencv-python-headless==4.7.0.72
pyclipper==1.3.0.post4
pycocoevalcap==1.2
pycocotools==2.0.6
rich==13.3.2
ruamel.yaml==0.17.21
scikit-learn==0.24.2
shapely==2.0.1
timm==0.6.12
transformers==4.26.1
yacs==0.1.8