一、查看CUDA版本
启动命令行:win+R,输入cmd,回车
在命令行中输入
nvidia-smi
输出看到自己的CUDA版本为11.1
二、安装pytorch1.4版本
因为conda安装低版本pytorch始终报错,因此直接使用pip下载。
pip install torch==1.4.0 torchvision==0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
-i https://pypi.tuna.tsinghua.edu.cn/simple/
三、数据集下载
1.命令行下载
wget http://mftp.mmcheng.net/liuyun/rcf/data/bsds_pascal_train_pair.lst
wget http://mftp.mmcheng.net/liuyun/rcf/data/HED-BSDS.tar.gz
wget http://mftp.mmcheng.net/liuyun/rcf/data/PASCAL.tar.gz
wget http://mftp.mmcheng.net/liuyun/rcf/data/NYUD.tar.gz
2.百度网盘(没有wget命令)
链接:https://pan.baidu.com/s/1BNopdAUOHf1qTGWnOlUb5g
提取码:8u8u
四、运行
根据README文件的提示,需要先运行train_RCF.py文件,直接通过命令行运行。
问题:如果出现ModuleNotFoundError: No module named 'XXX'
解决:通过命令行进行pip/conda下载。(网站很多这种类型,不一一举例)
问题:如果出现FileNotFoundError: [Errno 2] No such file or directory: 'data/HED-BSDS_PASCAL\\bsds_pascal_train_pair.lst错误,说明在指定文件夹位置没有找到相关文件
解决:需要改变文件位置。
问题:出现OMP ERROR
解决:在运行代码train_RCF添加
import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
问题:FileNotFoundError: [Errno 2] No such file or directory: 'vgg16convs.mat'
解决:1.下载网址https://drive.google.com/file/d/1lUhPKKj-BSOH7yQL0mOIavvrUbjydPp5/view
2.百度网盘下载:
链接:https://pan.baidu.com/s/1i1XAS3qCF0ThN6KvG-Y6HQ
提取码:npqf
运行成功如下(时间有点长得等)
参考:
1.解决报错:OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized
2.(Pytorch)环境配置与代码学习1—边缘检测:更丰富的卷积特征 Richer Convolutional Features for Edge Detection