# --extra-index-url https://download.pytorch.org/whl/cu113
# torch==1.11.0+cu113
# torchvision==0.12.0+cu113
torch
torchvision
numpy==1.21.6
opencv-python-headless>=4.2.0
scikit-image>=0.19.3
einops
tqdm
gradio
Pillow