运行GitHub代码--Windows（二），RCF-PyTorch运行train_RCF.py文件

一、查看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