《表3 注意力机制与语义关联分类模块对图像分类效果的影响》
为证明本文方法的有效性,在两个数据集上同时做了如表3所示的各模块的对比实验。由表可知,只使用denseNet模型进行分类时,最终的m AP结果在两个数据集上分别为89.9%和77.4%。在此基础之上,结合注意力机制,并利用一个SE模块来消除其产生的负响应,然后使用一个全局平均池化层和全连接层并与denseNet的输出融合,最终的mAP结果为90.0%和77.6%,这说明,只结合注意力机制对整体的分类效果影响不大,本文旨在利用注意力机制使得特征图通道和标签类别一一对应,并没有学习标签之间的关联性。不使用注意力机制,只采用denseNet和卷积与SE的结合,分别得到89.7%和77.0%的结果,相较于只使用denseNet的结果分别下降了0.2%和0.4%,这说明只增加网络层并不能促进分类效果的提升。最终的实验结果说明,利用注意力机制获取特征图通道和标签之间的对应关系,在此基础之上学习标签之间的依赖关系可以提升多标签图像分类结果。
图表编号 | XD0087180100 严禁用于非法目的 |
---|---|
绘制时间 | 2019.09.01 |
作者 | 薛丽霞、江迪、汪荣贵、杨娟 |
绘制单位 | 合肥工业大学计算机与信息学院、合肥工业大学计算机与信息学院、合肥工业大学计算机与信息学院、合肥工业大学计算机与信息学院 |
更多格式 | 高清、无水印(增值服务) |