如何使用Densenet/ResNet对训练、验证和测试图像进行多标签分类



我正在使用一个包含14720个训练图像的数据集进行多标签图像分类。我需要使用ResNet/DenseNet来实现它。我正在努力。你能给我推荐一位推荐人吗。

对于创建多标签分类问题,您必须记住两个不同的关键方面:

  1. 要使用的激活函数是sigmoid,而不是像在多类分类问题中那样的softmax
  2. 正确的标签将具有[1,0,1,0,0]的形式;实际上,由于我们有一个多标签,我们不存在互斥情况(事实上,这就是解释,关于伯努利分布,选择sigmoid而不是softmax在数学上更复杂(

您可以在这里了解如何在Keras:中创建多标签分类问题

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

最新更新