edge-connect 论文代码复现

1. 开始

1.1 论文的工作

1.2 论文的代码链接

2. 环境配置

2.1 自己电脑配置的环境

电脑配置： WIN11+RTX3060+CUDA11.3+python3.6+anaconda

Prerequisites

• Python 3
• PyTorch 1.0
• NVIDIA GPU + CUDA cuDNN

我的配置：Python3.6 pytorch1.10.2
对pytorch安装不熟悉的同学可以看看我的另一篇博客

requirement.txt （代码要求的库版本，在GitHub下载包里面有）

• numpy ~= 1.14.3
• scipy ~= 1.0.1
• future ~= 0.16.0
• matplotlib ~= 2.2.2
• pillow >= 6.2.0
• opencv-python ~= 3.4.0
• scikit-image ~= 0.14.0
• pyaml

2.2 其中遇到的问题

当时踩得坑：

• 当时用的是python3.7 配环境就各种报错，因为这上面大部分的库都很老，又加上论文是19年的，需要大家用anaconda
• 创造一个python低版本的环境。可是当我创造一个python3.5的版本，能正常安装requirement里面的库，但是又尴尬的发现，python3.5没有对应cuda11.3版本的pytorch，后面就换成python3.6
• python安装库时肯定会出现小问题，库版本正常调整就好

3. 测试（以数据集places2为例）

3.1 下载已经训练好的模型

3.2 测试文件配置

3.3 测试遇到常见问题

• 问题1：TypeError: load() missing 1 required positional argument: 'Loader'
  解决方案：看博客，修改对应的代码的位置

• 问题2：AttributeError: module 'scipy.misc' has no attribute 'imresize'
  解决方案：看博客2.2的部分，导入新的包，修改代码即可