我可以微调像Resnet和Vgg19这样的预训练分类CNN来预测图像中的某些角度吗?

时间:2018-07-23 01:39:18

标签: computer-vision conv-neural-network

最近,我在Blender中为一匹马建模,并随机更改其头部,颈部和腿部的旋转,并渲染了许多图像以构建数据集。

此后,我对resnet和vgg19进行了微调,以预测每个零件的旋转,但性能却很差。损失从0.9降到0.6,但没有达到足够低的值,并且验证集的性能非常差。

以下是我认为的一些潜在原因:

1不适用于此任务的微调分类模型。

2我略微更改了完全连接的层,并在其上添加了tanh以预测角度的正弦和余弦,然后最小化预测和地面真实性之间的均方误差。谭(Tanh)不适合执行此任务。

任何人都具有微调模型以预测角度的经验吗?

1 个答案:

答案 0 :(得分:0)

我没有微调模型来预测角度的经验。 但是对于我的对象检测任务经验,该任务可以完成。因为在物体检测中,某些输出是位置坐标。极坐标和直角坐标系可以通过矩阵乘法相互转换。因此,您可以尝试使用对象检测方法。