Open1

GSDT

PINTOPINTO
$ 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