回归网络的结果不佳

时间:2019-11-23 18:01:48

标签: matlab tensorflow deep-learning regression

我将NASNet(以及其他网络)用作回归网络。输出由4个实数组成。输入的是1536x1536图像,大小调整为331x331。

输出值描述多项式系数。前两个值比后两个值(1e-06至1e-02)小得多。

我第一次尝试使用125张训练图像和53张测试图像。

结果,网络输出值与期望值有很大差异。所有4个值都在1e-01和1e00范围内。

我该怎么做才能获得更好的结果?更多训练数据?避免调整图像大小?改变网络架构?我只用新的完全连接层(具有正确的输出编号)和回归层替换了分类和完全连接层。我认为问题在于回归层计算的是rmse,而不是相对的rmse。

0 个答案:

没有答案