使用预训练模型为什么门控网络输出的edge图片为黑色

DVLP-CMATERJU / RectiNet

A Gated and Bifurcated Stacked U-Net Module for Document Image Dewarping

GNU General Public License v3.0

101 stars 10 forks source link

Open hackerax opened 1 year ago

hackerax commented 1 year ago

我想了解门控网络的作用，使用预训练模型，发现门控网络输出为黑色图片，这里有什么错误吗？还是门控网络根本不起作用？

hackerax commented 1 year ago

save_image(normalize=True)函数加入normalize=True显示正确了，但是我这里输出的图片和论文中的不一致，论文中只包含边缘，同时我想知道作者是如何给edge打标签的？