Detail publikace

Fully Automated DCNN-Based Thermal Images Annotation Using Neural Network Pretrained on RGB Data

LIGOCKI, A. JELÍNEK, A. ŽALUD, L. RAHTU, E.

Originální název

Fully Automated DCNN-Based Thermal Images Annotation Using Neural Network Pretrained on RGB Data

Typ

článek v časopise ve Web of Science, Jimp

Jazyk

angličtina

Originální abstrakt

One of the biggest challenges of training deep neural network is the need for massive data annotation. To train the neural network for object detection, millions of annotated training images are required. However, currently, there are no large-scale thermal image datasets that could be used to train the state of the art neural networks, while voluminous RGB image datasets are available. This paper presents a method that allows to create hundreds of thousands of annotated thermal images using the RGB pre-trained object detector. A dataset created in this way can be used to train object detectors with improved performance. The main gain of this work is the novel method for fully automatic thermal image labeling. The proposed system uses the RGB camera, thermal camera, 3D LiDAR, and the pre-trained neural network that detects objects in the RGB domain. Using this setup, it is possible to run the fully automated process that annotates the thermal images and creates the automatically annotated thermal training dataset. As the result, we created a dataset containing hundreds of thousands of annotated objects. This approach allows to train deep learning models with similar performance as the common human-annotation-based methods do. This paper also proposes several improvements to fine-tune the results with minimal human intervention. Finally, the evaluation of the proposed solution shows that the method gives significantly better results than training the neural network with standard small-scale hand-annotated thermal image datasets.

Klíčová slova

deep convolutional neural networks; transfer learning; YOLO; RGB; IR; thermal; data annotation; object detector

Autoři

LIGOCKI, A.; JELÍNEK, A.; ŽALUD, L.; RAHTU, E.

Vydáno

24. 2. 2021

Nakladatel

MDPI

Místo

Postfach, CH-4020 Basel, Switzerland

ISSN

1424-8220

Periodikum

SENSORS

Ročník

21

Číslo

4

Stát

Švýcarská konfederace

Strany od

1

Strany do

23

Strany počet

23

URL

Plný text v Digitální knihovně

BibTex

@article{BUT170007,
  author="Adam {Ligocki} and Aleš {Jelínek} and Luděk {Žalud} and Esa {Rahtu}",
  title="Fully Automated DCNN-Based Thermal Images Annotation Using Neural Network Pretrained on RGB Data",
  journal="SENSORS",
  year="2021",
  volume="21",
  number="4",
  pages="1--23",
  doi="10.3390/s21041552",
  issn="1424-8220",
  url="https://www.mdpi.com/1424-8220/21/4/1552"
}