yun-liu / RCF

Richer Convolutional Features for Edge Detection
Other
761 stars 259 forks source link

输出图片都是黑白的 #150

Open lucy11111111 opened 2 years ago

lucy11111111 commented 2 years ago

您好,我想问一下,为什么我训练完以后,再测试输出的图片是这样的呢,每张图片大概有三张输出,第一张感觉像是六张图片拼在一起的,然后别的就都是黑白的,我只是简单的改了一下模型文件,为什么会这样讷

image

lucy11111111 commented 2 years ago

我用作者的源代码测试了一下,只要不使用vgg16的模型权重就会出现这样的结果,但是一旦涉及修改模型就无法使用预训练的权重啊,有什么解决办法吗

yun-liu commented 2 years ago

如果修改 Backbone 就不行,建议你别修改Backbone,修改之后的decoder。另一个思路是搞两个分支的网络,一个是vgg16,另一个分支你就可以随便搞,两个分支的特征交互、融合做预测,这样可能也能实现你改特征提取网络的计划。

lucy11111111 commented 2 years ago

嗯嗯好的,我还有一个问题,这两个avg分别代表什么呢?为什么训练的时候一直是下降的呢? image

Muzi010 commented 12 months ago

你好,楼主方便加个联系方式么?这环境我实在配不出来,可以加个联系方式说下正确流程么