迭代次数设置和加载模型

meteorshowers / RCF-pytorch

Richer Convolutional Features for Edge Detection model in pytorch CVPR2017

509 stars 145 forks source link

Open zhaoyuheng12 opened 4 years ago

zhaoyuheng12 commented 4 years ago

您好，最近对于RCF算法十分感兴趣，但是由于自身计算机能力较弱因此操作过程中遇到了很多困扰。请问训练模型时迭代次数在哪里能够控制；训练好的模型，应该输入到那个文件用来做边缘检测？

wangsun1996 commented 3 years ago

同学您好，我在训练的时候loss变化很大，并且最后两层出来的图都是全黑的。我想要用RCF模型进行一些图像边缘的检测，请问您方便给我分享一下您训练好的模型文件吗？

cainiaoqiuzhu commented 3 years ago

同学您好，我在训练的时候loss变化很大，并且最后两层出来的图都是全黑的。我想要用RCF模型进行一些图像边缘的检测，请问您方便给我分享一下您训练好的模型文件吗？

我也是遇到这种情况，我也觉得很奇怪