1、配置环境
anaconda创建一个新环境STDN
conda create -n STDN python=3.6
conda activate STDN
启动STDN环境
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
安装pyrorch
新建一个文件夹,存放STDN程序
git clone https://github.com/arvention/STDN
2、修改文件
https://blog.csdn.net/xiao_lxl/article/details/85047285
参照上面连接
python3 main.py
运行,报错
1:No module named 'tqdm' 'cv2'
conda install tqdm
pip isntall opencv-python
2: VOC的格式