无需像素标注实现物体共分割,提升手绘草图分类效果
Co-Segmentation without any Pixel-level Supervision with Application to Large-Scale Sketch Classification
- 用两个预训练ViT模型估算图像内和类间特征相关性,实现无像素监督的共分割
- 在挑战性基准上达到与有像素标注方法相当的性能,且仅需图像标签
- 利用共分割区域提取草图特征,显著提升大规模手绘草图分类效果
本文提出一种无需像素级监督的物体共分割新方法,通过两个预训练Vision Transformer(ViT)模型实现:基于ImageNet分类训练的ViT用于估计类内标记相关性以获得粗略目标定位,而自监督DINO-ViT用于捕捉图像内标记相关性。在多个近期挑战性基准上,该方法在仅使用图像标签的监督水平下达到当前最优性能,并与使用二值掩码进行像素级监督的方法相媲美。进一步验证了该共分割方法在大规模手绘草图识别任务中的优势:通过利用大量自然图像的图像级标注数据,结合从自然图像边缘提取的草图相似代理域来缓解领域差距。实验表明,当分类器在共分割区域提取的草图结构上训练时,草图识别性能显著优于全图训练。
原文摘要 · Abstract (English)
This work proposes a novel method for object co-segmentation, i.e. pixel-level localization of a common object in a set of images, that uses no pixel-level supervision for training. Two pre-trained Vision Transformer (ViT) models are exploited: ImageNet classification-trained ViT, whose features are used to estimate rough object localization through intra-class token relevance, and a self-supervised DINO-ViT for intra-image token relevance. On recent challenging benchmarks, the method achieves state-of-the-art performance among methods trained with the same level of supervision (image labels) while being competitive with methods trained with pixel-level supervision (binary masks). The benefits of the proposed co-segmentation method are further demonstrated in the task of large-scale sketch recognition, that is, the classification of sketches into a wide range of categories. The limited amount of hand-drawn sketch training data is leveraged by exploiting readily available image-level-annotated datasets of natural images containing a large number of classes. To bridge the domain gap, the classifier is trained on a sketch-like proxy domain derived from edges detected on natural images. We show that sketch recognition significantly benefits when the classifier is trained on sketch-like structures extracted from the co-segmented area rather than from the full image. Code: https://github.com/nikosips/CBNC .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。