IEEE Trans Image Process. 2023;32:3000-3012. doi: 10.1109/TIP.2023.3266161. Epub 2023 May 26.
Multi-label image classification is a fundamental but challenging task in computer vision. To tackle the problem, the label-related semantic information is often exploited, but the background context and spatial semantic information of related objects are not fully utilized. To address these issues, a multi-branch deep neural network is proposed in this paper. The first branch is designed to extract the discriminant information from regions of interest to detect target objects. In the second branch, a spatial context-aware approach is proposed to better capture the contextual information of an object in its surroundings by using an adaptive patch expansion mechanism. It helps the detection of small objects that are easily lost without the support of context information. The third one, the object-attentional branch, exploits the spatial semantic relations between the target object and its related objects, to better detect partially occluded, small or dim objects with the support of those easily detectable objects. To better encode such relations, an attention mechanism jointly considering the spatial and semantic relations between objects is developed. Two widely used benchmark datasets for multi-labeling classification, MS COCO and PASCAL VOC, are used to evaluate the proposed framework. The experimental results demonstrate that the proposed method outperforms the state-of-the-art methods for multi-label image classification.
多标签图像分类是计算机视觉中的一项基本但具有挑战性的任务。为了解决这个问题,通常会利用与标签相关的语义信息,但相关对象的背景上下文和空间语义信息并未得到充分利用。针对这些问题,本文提出了一种多分支深度神经网络。第一分支旨在从感兴趣区域中提取判别信息,以检测目标对象。第二分支提出了一种空间上下文感知方法,通过自适应补丁扩展机制更好地捕获对象周围的上下文信息。这有助于检测到没有上下文信息支持很容易丢失的小目标。第三分支,即目标注意分支,利用目标对象与其相关对象之间的空间语义关系,在那些容易检测到的对象的支持下更好地检测部分遮挡、小或暗淡的对象。为了更好地编码这些关系,开发了一种同时考虑对象之间空间和语义关系的注意力机制。使用两个广泛用于多标签分类的基准数据集,即 MS COCO 和 PASCAL VOC,来评估所提出的框架。实验结果表明,所提出的方法在多标签图像分类方面优于最新方法。