用于预测两个图像之间旋转的强化学习

时间:2018-05-23 03:22:01

标签: python keras deep-learning reinforcement-learning

我有相对于彼此随机旋转的成对图像。我想预测旋转角度。我不知道真正的旋转角度,但我知道在应用正确的旋转后两个图像之间的MSE应该最小化。

我可以生成具有已知轮换和训练网络的训练数据。 但是我们可以在没有训练数据的情况下做到这一点,原始图像和旋转图像之间的MSE作为损失函数吗?

我尝试添加一个图层,将预测旋转应用于一个输入图像,并与第二个输入图像进行比较。 或者,我们可以使用自定义损失函数来执行此操作,但它需要将输入图像和预测旋转作为输入,而不是y_pred和y_true。

但我不知道如何在Keras做到这一点。我只是一个新手,所以非常感谢帮助。

1 个答案:

答案 0 :(得分:0)

如果您已经了解神经网络的工作方式以及python的一些经验,我建议您先了解更多关于keras的知识。 This tutorial是开始使用keras并学习它能力的一个很好的资源。这应该让你开始朝着正确的方向前进,然后当你有更具体的问题要求我们能够更有效地帮助你时。