Open ouyangbetter opened 4 months ago
conda create -n sta_env python=3.10 conda activate sta_env
git clone https://github.com/z-x-yang/Segment-and-Track-Anything.git cd Segment-and-Track-Anything
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
cd sam pip install -e . cd ..
pip install -e git+https://github.com/IDEA-Research/GroundingDINO.git@main#egg=GroundingDINO
pip install numpy opencv-python pycocotools matplotlib scikit-image
pip install Pillow==9.5.0 pip install supervision
pip install gradio==3.39.0 gdown ffmpeg==1.4
git clone https://github.com/ClementPinard/Pytorch-Correlation-extension.git cd Pytorch-Correlation-extension pip install -e . cd ..
mkdir ckpt cd ckpt
cd ..
python app.py
(wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run ) (sh cuda_11.8.0_520.61.05_linux.run --silent --toolkit --override --installpath=$HOME/cuda ) (export CUDA_HOME=$HOME/cuda export PATH=$CUDA_HOME/bin:$PATH export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH ) (echo 'export CUDA_HOME=$HOME/cuda' >> ~/.bashrc echo 'export PATH=$CUDA_HOME/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc )
/home/user/repos/Segment-and-Track-Anything/src/groundingdino/groundingdino/models/GroundingDINO/ms_deform_attn.py:31: UserWarning: Failed to load custom C++ ops. Running on CPU mode Only!