DeepSnake环境配置

DeepSnake

文章：https://arxiv.org/abs/2001.01629

代码：GitHub - zju3dv/snake: Code for "Deep Snake for Real-Time Instance Segmentation" CVPR 2020 oral

配置环境

本地环境：Linux(Ubuntu 21.10)+cuda11.3

配置环境：按照官方提供的install.md进行安装

1.安装pytorch

conda create -n snake python=3.7
conda activate snake
conda install pytorch==1.10.2 torchvision==0.11.3 cudatoolkit=11.3 -c pytorch -c conda-forge

2.安装所需包

pip install Cython yacs numpy tqdm opencv-contrib-python==3.4.2.17 opencv-python==3.4.2.17 imgaug pycocotools Pillow tensorboardX termcolor

3.安装apex

git clone https://github.com/NVIDIA/apex.git
cd a