草庐IT

EdgeConnect

全部标签

计算机视觉大作业:EdgeConnect论文阅读

复现一、环境配置运行环境:Python3.6.6PyTorch1.10.0NVIDIAGPU+CUDA10.2+cuDNN8.2Python包:numpy=1.14.3+mklscipy=1.0.1future=0.18.2matplotlib=2.2.2pillow=8.3.1opencv-python=4.5.3.56scikit-image=0.14.5Pyaml=21.10.1二、下载预训练模型三、运行模型(若要自己训练模型,要提供原图的边缘检测结果、掩膜和原图灰度图,再加上训练时间可能较长,我们就直接采用作者训练好的模型去跑结果了。。。)Windows10下,在命令行输入:pytho