keras 已知对象数目和对象形状的语义分割损失函数

b4wnujal  于 2022-11-13  发布在  其他
关注(0)|答案(1)|浏览(160)

在我的用例中,对象的数量总是一个,并且对象总是椭圆形的。在训练时,我注意到分割模型偶尔会预测多个对象(噪声是其他的)。
有没有办法用公式表示一个Keras损失函数来奖励接近椭圆质量中心的预测,而惩罚远离中心的预测,同时保持椭圆形状?
如何实现一个损失函数来处理这样的限制呢?我知道有一些损失函数,比如多个对象的wasserstein损失,当已知的对象层次结构不满足时,就会受到惩罚。

kninwzqo

kninwzqo1#

如果始终只有一个对象,则可以尝试将其视为回归问题,并预测椭圆的参数。例如:中心(x,y)、轴长度和Angular 。
https://en.wikipedia.org/wiki/Ellipse#General_ellipse

相关问题