我正在使用一个包含14720个训练图像的数据集进行多标签图像分类。我需要使用ResNet/DenseNet来实现它。我正在努力。你能给我推荐一位推荐人吗。
对于创建多标签分类问题,您必须记住两个不同的关键方面:
- 要使用的激活函数是
sigmoid
,而不是像在多类分类问题中那样的softmax
- 正确的标签将具有
[1,0,1,0,0]
的形式;实际上,由于我们有一个多标签,我们不存在互斥情况(事实上,这就是解释,关于伯努利分布,选择sigmoid而不是softmax在数学上更复杂(
您可以在这里了解如何在Keras:中创建多标签分类问题
https://www.pyimagesearch.com/2018/05/07/multi-label-classification-with-keras/