【发布时间】:2018-12-07 06:28:45
【问题描述】:
我正在训练一个边界框回归问题,并使用YOLO paper 中描述的损失函数的坐标分量。
方法 1:使用 HOG 特征向量作为 ANN 的输入。相同的损失函数。给我一个好模型。
方法 2:使用图像作为 CNN 的输入。最后一层与方法 1 相同。损失函数也相同。但是我面临着计算损失的问题。从第一个纪元本身开始,它就打印出用于训练、开发和测试的 nan 损失。
当相同的损失函数与 ANN 完美配合时,我该如何让 CNN 工作?发生这种情况的原因是什么?我希望能够使用相同的损失函数。
【问题讨论】:
-
这还有问题吗?
标签: python tensorflow nan convolutional-neural-network