Saved in:
Bibliographic Details
Main Authors: Mohwald, Albert, Jenicek, Tomas, Chum, Ondřej
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2309.16351
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914881031110656
author Mohwald, Albert
Jenicek, Tomas
Chum, Ondřej
author_facet Mohwald, Albert
Jenicek, Tomas
Chum, Ondřej
contents Image retrieval methods based on CNN descriptors rely on metric learning from a large number of diverse examples of positive and negative image pairs. Domains, such as night-time images, with limited availability and variability of training data suffer from poor retrieval performance even with methods performing well on standard benchmarks. We propose to train a GAN-based synthetic-image generator, translating available day-time image examples into night images. Such a generator is used in metric learning as a form of augmentation, supplying training data to the scarce domain. Various types of generators are evaluated and analyzed. We contribute with a novel light-weight GAN architecture that enforces the consistency between the original and translated image through edge consistency. The proposed architecture also allows a simultaneous training of an edge detector that operates on both night and day images. To further increase the variability in the training examples and to maximize the generalization of the trained model, we propose a novel method of diverse anchor mining. The proposed method improves over the state-of-the-art results on a standard Tokyo 24/7 day-night retrieval benchmark while preserving the performance on Oxford and Paris datasets. This is achieved without the need of training image pairs of matching day and night images. The source code is available at https://github.com/mohwald/gandtr .
format Preprint
id arxiv_https___arxiv_org_abs_2309_16351
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Dark Side Augmentation: Generating Diverse Night Examples for Metric Learning
Mohwald, Albert
Jenicek, Tomas
Chum, Ondřej
Computer Vision and Pattern Recognition
Image retrieval methods based on CNN descriptors rely on metric learning from a large number of diverse examples of positive and negative image pairs. Domains, such as night-time images, with limited availability and variability of training data suffer from poor retrieval performance even with methods performing well on standard benchmarks. We propose to train a GAN-based synthetic-image generator, translating available day-time image examples into night images. Such a generator is used in metric learning as a form of augmentation, supplying training data to the scarce domain. Various types of generators are evaluated and analyzed. We contribute with a novel light-weight GAN architecture that enforces the consistency between the original and translated image through edge consistency. The proposed architecture also allows a simultaneous training of an edge detector that operates on both night and day images. To further increase the variability in the training examples and to maximize the generalization of the trained model, we propose a novel method of diverse anchor mining. The proposed method improves over the state-of-the-art results on a standard Tokyo 24/7 day-night retrieval benchmark while preserving the performance on Oxford and Paris datasets. This is achieved without the need of training image pairs of matching day and night images. The source code is available at https://github.com/mohwald/gandtr .
title Dark Side Augmentation: Generating Diverse Night Examples for Metric Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2309.16351