配置GlobalTrack到Anti-UAV(一)

论文:GlobalTrack: A Simple and Strong Baseline for Long-term Tracking

代码地址:https://github.com/huanglianghua/GlobalTrack

1、下载代码,创建虚拟环境并激活

2、安装pytorch(1.10.1)和cuda(11.3)

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

3、安装多个包,包括:imageio numpy opencv-python opencv-python-headless Pillow scikit-image scikit-learn scipy Shapely matplotlib pycocotools terminaltables

4、安装mmcv(安装mmcv-full后面编译会出错)

pip install mmcv==0.4.3 (高版本和自带的mmdetection会冲突出错)

5、编译mmdetection

cd _submodules/mmdetection/

python setup.py develop

如果出现错误:RuntimeError: Error compiling objects for extension,则将项目中所有的AT_CHECK替换为TORCH_CHECK(注意大小写)。并且执行rm -rf build .eggs mmdet.egg-info来删除之前编译的信息

再次运行python setup.py develop即可成功(据说用conda安装pytorch 1.8.1 & CUDA 11.1时这里会报错,需要用pip,本人未证实)

你可能感兴趣的:(python,深度学习,pytorch)