OneFormer / requirements.txt
praeclarumjj3's picture
Fix req.txt
80a2093
raw
history blame
604 Bytes
--extra-index-url https://download.pytorch.org/whl
torch==1.9.0+cu111
torchvision==0.10.0+cu111
torchaudio==0.9.0
-f https://dl.fbaipublicfiles.com/detectron2/wheels/cu111/torch1.9/index.html
detectron2==0.6
opencv-python
git+https://github.com/cocodataset/panopticapi.git
git+https://github.com/mcordts/cityscapesScripts.git
imutils
-f https://shi-labs.com/natten/wheels/cu111/torch1.9/index.html
natten==0.14.2
cython
scipy
shapely
h5py
submitit
scikit-image
timm==0.4.12
einops
icecream
setuptools==59.5.0
wandb
ftfy
regex
inflect
diffdist
gdown
huggingface_hub
pytorch_lightning
tqdm
gradio==3.1.4