Open1
GSDT
$ docker run --gpus all --rm -it pytorch/pytorch:1.4-cuda10.1-cudnn7-devel
# apt update \
&& apt upgrade -y \
&& apt install -y python3-pip libgl1-mesa-dev \
&& pip install pip --upgrade \
&& pip install cython \
&& pip install numpy \
&& pip install cython-bbox \
&& pip install ffmpeg \
&& pip install gdown \
&& pip install opencv-contrib-python \
&& pip install torch_sparse
# git clone https://github.com/yongxinw/GSDT.git
# cd GSDT
# pip install -r requirements.txt
# bash install_pyg.sh cu101
# cd ./src/lib/models/networks/DCNv2
# ln -s /usr/local/cuda-10.2/targets/x86_64-linux/include/cublas_v2.h \
/usr/local/cuda-10.1/targets/x86_64-linux/include/cublas_v2.h
# ln -s /usr/local/cuda-10.2/targets/x86_64-linux/include/cublas_api.h \
/usr/local/cuda-10.1/targets/x86_64-linux/include/cublas_api.h
# bash make.sh
# cd /GSDT/experiments
# gdown --id 1K_6yN1jD7fpmGN23Z4NgsROO5mZRf6ay
# chmod +x track_gnn_mot_AGNNConv_RoIAlign_mot15.sh
# track_gnn_mot_AGNNConv_RoIAlign_mot15.sh model_mot15.pth