Spaces:
Running
Running
File size: 1,132 Bytes
32faf2b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
# create new conda env
conda create -n SAA python=3.9
source activate SAA
# PyTorch
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
# $ProjectRoot: the root you save our project, e.g., /home/anyad/VAND-solution
ProjectRoot=/home/anyad/VAND-solution
cd $ProjectRoot
# SAM and DINO
cd ./GroundingDINO
pip install -e .
cd ../SAM
pip install -e .
pip install setuptools==59.5.0
pip install --upgrade diffusers[torch]
pip install opencv-python pycocotools matplotlib onnxruntime onnx ipykernel
pip install transformers
pip install addict
pip install yapf
pip install timm
pip install loguru
pip install tqdm
pip install scikit-image
pip install scikit-learn
pip install pandas
pip install tensorboard
pip install seaborn
pip install open_clip_torch
pip install SciencePlots
pip install timm
pip install einops
pip install gradio
# weights
cd ../
mkdir weights
cd ./weights/
wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
wget https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
|