ODISE / requirements.txt
cqys's picture
Update requirements.txt
8171eac verified
raw
history blame contribute delete
368 Bytes
--extra-index-url https://download.pytorch.org/whl/cpu
torch==1.13.1
torchvision==0.14.1
xformers==0.0.16
numpy==1.23.5
matplotlib==3.7.1
pillow==9.4.0
gradio==3.29.0
datasets==4.5.0
transformers==4.35.0
# Git 依赖
git+https://github.com/NVlabs/ODISE.git
git+https://github.com/facebookresearch/Mask2Former.git
git+https://github.com/facebookresearch/detectron2.git