如何利用Densenet/ResNet实现多标签分类

Question

我正在工作的多标签图像分类使用有14,720个培训图像的数据集.我需要使用ResNet/DenseNet实现它。我试着解决这个问题。你能建议我参考一下前进的路吗？

Answer 1

要创建多标签分类问题，您必须记住两个不同的关键方面：

对于Bernoulli Distributions).，我们使用的激活函数是sigmoid，而不是softmax，就像在多类分类问题中那样。

• 一个正确的标签将是形式[1,0,1,0,0]；实际上，由于我们有一个多标签，所以我们没有相互排斥的情况(实际上，这是一种更复杂的数学解释，对于Bernoulli

• 来说，不是softmax )。

您可以在这里查看如何在Keras中创建多标签分类问题：

https://www.pyimagesearch.com/2018/05/07/multi-label-classification-with-keras/