如何将单个图像的排序的多个标签馈送到有监督的训练管道中

问题描述 投票:0回答:1

在用于图像分类的监督学习中,我们在训练图像中添加标签。 是否可以为每个图像提供一系列按相关性顺序排列的标签,而不仅仅是一个标签?]] >>

我正在使用PyTorch进行此有监督的培训。现在,在每个标签下,我都有该标签的训练图像。谁能帮助我说出如何在PyTorch中实施分级的多标签图像训练?

在用于图像分类的监督学习中,我们在训练图像中添加标签。是否可以为每个图像提供一系列按相关性顺序排列的标签,而不是...

image pytorch classification supervised-learning
1个回答
0
投票

更具体的示例将有助于更好地了解您的问题。但是据我了解,您并没有解决经典的分类问题,我们将使用CrossEntropyLoss,但您想学习一种相关性度量。

我知道的最好的方法是使用度量学习(学习排名是一个似乎适合您的问题的示例)。 Triplet Learning特别著名,但是还有其他方法可以使之成为可能。您可能需要在这里查看:https://gombru.github.io/2019/04/03/ranking_loss/,在这里:https://towardsdatascience.com/image-similarity-using-triplet-loss-3744c0f67973

© www.soinside.com 2019 - 2024. All rights reserved.