While studying objects presented in paintings, art history specialists identify their significance, symbolic meaning and historical context. Analyzing big artistic collections can be very time-consuming for the specialists. The search could be relieved by using modern object detectors. However, object detectors have poor performance on artistic images. This problem could be solved by fine-tuning them on specialized annotated datasets. In this paper, we explore the possibilities of using open-vocabulary foundation models for dataset annotation in a semi-automated manner. We propose an approach for artistic dataset annotation for object detection task based on a small set of images annotated on image-level and using Vision Transformer for Open-World Localization (OWL-ViT2) model, the YOLO object detector and an approximate nearest neighbour oh yeah (ANNOY) algorithm. We extend the existing DEArt dataset by 97.2 \(\%\) and introduce the way of adding new classes without exhaustive annotation. With the extended version of the dataset, we achieve 12.2 \(\%\) increase of mAP0.5 metric on average on the test data compared to the model trained on the original dataset.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

An Approach for Dataset Extension for Object Detection in Artworks Using Open-Vocabulary Models

  • Tetiana Yemelianenko,
  • Iuliia Tkachenko,
  • Tess Masclef,
  • Mihaela Scuturici,
  • Serge Miguet

摘要

While studying objects presented in paintings, art history specialists identify their significance, symbolic meaning and historical context. Analyzing big artistic collections can be very time-consuming for the specialists. The search could be relieved by using modern object detectors. However, object detectors have poor performance on artistic images. This problem could be solved by fine-tuning them on specialized annotated datasets. In this paper, we explore the possibilities of using open-vocabulary foundation models for dataset annotation in a semi-automated manner. We propose an approach for artistic dataset annotation for object detection task based on a small set of images annotated on image-level and using Vision Transformer for Open-World Localization (OWL-ViT2) model, the YOLO object detector and an approximate nearest neighbour oh yeah (ANNOY) algorithm. We extend the existing DEArt dataset by 97.2 \(\%\) and introduce the way of adding new classes without exhaustive annotation. With the extended version of the dataset, we achieve 12.2 \(\%\) increase of mAP0.5 metric on average on the test data compared to the model trained on the original dataset.